U mdj+@szddlZddlmZmZddlmZmZmZddlm Z GdddZ Gd d d Z Gd d d eeZ Gd ddeZ dS)N) BaseEstimatorClassifierMixin)_check_sample_weight _num_samples check_array)check_is_fittedc@s eZdZdZddZddZdS)ArraySlicingWrapper- Parameters ---------- array cCs ||_dSNarrayselfrrO/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/sklearn/utils/_mocking.py__init__szArraySlicingWrapper.__init__cCst|j|Sr  MockDataFramer)rZaslicerrr __getitem__szArraySlicingWrapper.__getitem__N)__name__ __module__ __qualname____doc__rrrrrrr sr c@sDeZdZdZddZddZdddZd d Zd d ZdddZ dS)rr cCs*||_||_|j|_|j|_t||_dSr )rvaluesshapendimr Zilocrrrrrs zMockDataFrame.__init__cCs t|jSr )lenrrrrr__len__'szMockDataFrame.__len__NcCs|jSr r )rdtyperrr __array__*szMockDataFrame.__array__cCst|j|jkSr rrotherrrr__eq__0szMockDataFrame.__eq__cCs ||k Sr rr#rrr__ne__3szMockDataFrame.__ne__rcCst|jj||dS)N)axis)rrtake)rindicesr'rrrr(6szMockDataFrame.take)N)r) rrrrrr r"r%r&r(rrrrrs rc @sjeZdZdZdddddddddddZdd d Zdd d Zd dZddZddZ dddZ ddZ dS)CheckingClassifieraDummy classifier to test pipelining and meta-estimators. Checks some property of `X` and `y`in fit / predict. This allows testing whether pipelines / cross-validation or metaestimators changed the input. Can also be used to check if `fit_params` are passed correctly, and to force a certain score to be returned. Parameters ---------- check_y, check_X : callable, default=None The callable used to validate `X` and `y`. These callable should return a bool where `False` will trigger an `AssertionError`. check_y_params, check_X_params : dict, default=None The optional parameters to pass to `check_X` and `check_y`. methods_to_check : "all" or list of str, default="all" The methods in which the checks should be applied. By default, all checks will be done on all methods (`fit`, `predict`, `predict_proba`, `decision_function` and `score`). foo_param : int, default=0 A `foo` param. When `foo > 1`, the output of :meth:`score` will be 1 otherwise it is 0. expected_sample_weight : bool, default=False Whether to check if a valid `sample_weight` was passed to `fit`. expected_fit_params : list of str, default=None A list of the expected parameters given when calling `fit`. Attributes ---------- classes_ : int The classes seen during `fit`. n_features_in_ : int The number of features seen during `fit`. Examples -------- >>> from sklearn.utils._mocking import CheckingClassifier This helper allow to assert to specificities regarding `X` or `y`. In this case we expect `check_X` or `check_y` to return a boolean. >>> from sklearn.datasets import load_iris >>> X, y = load_iris(return_X_y=True) >>> clf = CheckingClassifier(check_X=lambda x: x.shape == (150, 4)) >>> clf.fit(X, y) CheckingClassifier(...) We can also provide a check which might raise an error. In this case, we expect `check_X` to return `X` and `check_y` to return `y`. >>> from sklearn.utils import check_array >>> clf = CheckingClassifier(check_X=check_array) >>> clf.fit(X, y) CheckingClassifier(...) Nallrcheck_ycheck_y_paramscheck_Xcheck_X_paramsmethods_to_check foo_paramexpected_sample_weightexpected_fit_paramsc Cs4||_||_||_||_||_||_||_||_dSr r,) rr-r.r/r0r1r2r3r4rrrrzs zCheckingClassifier.__init__TcCs|r t||jdk rV|jdkr$in|j}|j|f|}t|ttjfrR|sVtn|}|dk r|jdk r|j dkrvin|j }|j|f|}t|ttjfr|stn|}||fS)aValidate X and y and make extra check. Parameters ---------- X : array-like of shape (n_samples, n_features) The data set. y : array-like of shape (n_samples), default=None The corresponding target, by default None. should_be_fitted : bool, default=True Whether or not the classifier should be already fitted. By default True. Returns ------- X, y N) r r/r0 isinstanceboolnpZbool_AssertionErrorr-r.)rXyshould_be_fittedparamsZ checked_XZ checked_yrrr _check_X_ys   zCheckingClassifier._check_X_yc Kst|t|kst|jdks(d|jkr<|j||dd\}}t|d|_tt|ddd|_ |j rt |j t |}|rtdt |d | D]<\}}t|t|krtd |d t|d t|d q|jr|dkrtdt|||S)aFit classifier. Parameters ---------- X : array-like of shape (n_samples, n_features) Training vector, where `n_samples` is the number of samples and `n_features` is the number of features. y : array-like of shape (n_samples, n_outputs) or (n_samples,), default=None Target relative to X for classification or regression; None for unsupervised learning. sample_weight : array-like of shape (n_samples,), default=None Sample weights. If None, then samples are equally weighted. **fit_params : dict of string -> object Parameters passed to the ``fit`` method of the estimator Returns ------- self r+fitF)r;rT)Z ensure_2dZallow_ndzExpected fit parameter(s) z not seen.zFit parameter z has length z ; expected .Nz#Expected sample_weight to be passed)rr8r1r=r7rZn_features_in_uniquerclasses_r4setlistitemsr3r)rr9r:Z sample_weightZ fit_paramsmissingkeyvaluerrrr>s* zCheckingClassifier.fitcCs:|jdksd|jkr"||\}}|jtjt|tdS)a>Predict the first class seen in `classes_`. Parameters ---------- X : array-like of shape (n_samples, n_features) The input data. Returns ------- preds : ndarray of shape (n_samples,) Predictions of the first class seens in `classes_`. r+predict)r!)r1r=rAr7zerosrintrr9r:rrrrHs zCheckingClassifier.predictcCsN|jdksd|jkr"||\}}tt|t|jf}d|dddf<|S)aPredict probabilities for each class. Here, the dummy classifier will provide a probability of 1 for the first class of `classes_` and 0 otherwise. Parameters ---------- X : array-like of shape (n_samples, n_features) The input data. Returns ------- proba : ndarray of shape (n_samples, n_classes) The probabilities for each sample and class. r+ predict_probarNr)r1r=r7rIrrrA)rr9r:ZprobarrrrLs z CheckingClassifier.predict_probacCsn|jdksd|jkr"||\}}t|jdkr>tt|Stt|t|jf}d|dddf<|SdS)aBConfidence score. Parameters ---------- X : array-like of shape (n_samples, n_features) The input data. Returns ------- decision : ndarray of shape (n_samples,) if n_classes == 2 else (n_samples, n_classes) Confidence score. r+decision_functionrrNr)r1r=rrAr7rIr)rr9r:ZdecisionrrrrM sz$CheckingClassifier.decision_functioncCs8|jdksd|jkr ||||jdkr0d}nd}|S)aQFake score. Parameters ---------- X : array-like of shape (n_samples, n_features) Input data, where `n_samples` is the number of samples and `n_features` is the number of features. Y : array-like of shape (n_samples, n_output) or (n_samples,) Target relative to X for classification or regression; None for unsupervised learning. Returns ------- score : float Either 0 or 1 depending of `foo_param` (i.e. `foo_param > 1 => score=1` otherwise `score=0`). r+scorerg?g)r1r=r2)rr9YrNrrrrN%s   zCheckingClassifier.scorecCs ddgdS)NTZ1dlabel) _skip_testZX_typesrrrrr _more_tags@szCheckingClassifier._more_tags)NT)N)NN) rrrrrr=r>rHrLrMrNrQrrrrr*:s"B  # 0 r*c@s:eZdZdZd ddZddZddZd d Zd d ZdS)NoSampleWeightWrapperzWrap estimator which will not expose `sample_weight`. Parameters ---------- est : estimator, default=None The estimator to wrap. NcCs ||_dSr )est)rrSrrrrMszNoSampleWeightWrapper.__init__cCs|j||Sr )rSr>rKrrrr>PszNoSampleWeightWrapper.fitcCs |j|Sr )rSrHrr9rrrrHSszNoSampleWeightWrapper.predictcCs |j|Sr )rSrLrTrrrrLVsz#NoSampleWeightWrapper.predict_probacCsddiS)NrPTrrrrrrQYsz NoSampleWeightWrapper._more_tags)N) rrrrrr>rHrLrQrrrrrRDs  rR)numpyr7baserrZ validationrrrr r rr*rRrrrrs $