rtj5ddlZddlmZmZddlmZddlmZddl m Z m Z m Z m Z mZGddZGdd ZGd d eeZed gd e_GddeZdZGddeZdS)N) BaseEstimatorClassifierMixin) RequestMethod) available_if)_check_sample_weight _num_samples check_arraycheck_is_fittedcheck_random_stateceZdZdZdZdZdS)ArraySlicingWrapper- Parameters ---------- array c||_dSNarrayselfrs T/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sklearn/utils/_mocking.py__init__zArraySlicingWrapper.__init__s  c6t|j|Sr MockDataFramer)raslices r __getitem__zArraySlicingWrapper.__getitem__sTZ/000rN)__name__ __module__ __qualname____doc__rrrrr r s< 11111rr c:eZdZdZdZdZd dZdZdZd d Z dS) rrcz||_||_|j|_|j|_t ||_dSr)rvaluesshapendimr ilocrs rrzMockDataFrame.__init__)s5  [ J '.. rc*t|jSr)lenr)rs r__len__zMockDataFrame.__len__1s4:rNc|jSrr)rdtypes r __array__zMockDataFrame.__array__4s zrc<t|j|jkSrrrothers r__eq__zMockDataFrame.__eq__:sTZ5;6777rc||k Srr!r/s r__ne__zMockDataFrame.__ne__=s5=  rrcTt|j||S)Naxis)rrtake)rindicesr6s rr7zMockDataFrame.take@s"TZ__W4_@@AAArr)r) rrrr rr*r-r1r3r7r!rrrr s/// 888!!!BBBBBBrrc heZdZdZdddddddddd dZddZdd Zd Zd Zd Z dd Z fdZ xZ S)CheckingClassifiera$ Dummy 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`. If `None`, the data is not validated. Default is `None`. check_y_params, check_X_params : dict, default=None The optional parameters to pass to `check_X` and `check_y`. If `None`, then no parameters are passed in. 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(...) Nallr check_ycheck_y_paramscheck_Xcheck_X_paramsmethods_to_check foo_paramexpected_sample_weightexpected_fit_params random_statec ||_||_||_||_||_||_||_||_| |_dSrr<) rr=r>r?r@rArBrCrDrEs rrzCheckingClassifier.__init__sO , , 0"&<##6 (rTcd|rt||jF|jin|j}|j|fi|}t|tt jfr|sJn|}|M|jF|jin|j}|j|fi|}t|tt jfr|sJn|}||fS)atValidate X and y and make extra check. Parameters ---------- X : array-like of shape (n_samples, n_features) The data set. `X` is checked only if `check_X` is not `None` (default is None). y : array-like of shape (n_samples), default=None The corresponding target, by default `None`. `y` is checked only if `check_y` is not `None` (default is None). should_be_fitted : bool, default=True Whether or not the classifier should be already fitted. By default True. Returns ------- X, y ) r r?r@ isinstanceboolnpbool_r=r>)rXyshould_be_fittedparams checked_X checked_ys r _check_X_yzCheckingClassifier._check_X_ys&  " D ! ! ! < #.6RRD object Parameters passed to the ``fit`` method of the estimator Returns ------- self r;fitF)rNT) ensure_2dallow_ndzExpected fit parameter(s) z not seen.zFit parameter z has length z ; expected .Nz#Expected sample_weight to be passed)rrArRrJr%n_features_in_uniquer classes_rDsetAssertionErrorlistitemsrCr)rrLrM sample_weight fit_paramsmissingkeyvalues rrTzCheckingClassifier.fits0A,q//1111  E ) )Ud6K-K-K??1a%?@@DAq hqkk!n +a54"P"P"PQQ  # $233c*ooEG $JgJJJ)..00   U&&,q//99(999,u:M:M99&21oo999:  & 3$$%JKKK  2 2 2 rc|jdks d|jvr||\}}t|j}||jt |S)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 seen in `classes_`. r;predict)size)rArRr rEchoicer[rrrLrMrngs rrfzCheckingClassifier.predictsb  E ) )Y$:O-O-O??1%%DAq !233zz$-l1ooz>>>rcv|jdks d|jvr||\}}t|j}|t |t |j}tj ||}|tj |dddtj fz}|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_proba)outrUr5N) rArRr rErandnrr)r[rJabssumnewaxis)rrLrMrjprobas rrlz CheckingClassifier.predict_probas  E ) )_@U-U-U??1%%DAq !233 ,q//3t}+=+=>>u%((( A&&&qqq"*}55 rc`|jdks d|jvr||\}}t|j}t |jdkr"|t|S|t|t |jS)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_function)rArRr rEr)r[rnrris rrtz$CheckingClassifier.decision_functions  !U * *"d&;;;??1%%DAq !233 t}   " "99\!__-- -99\!__c$-.@.@AA Arcz|jdks d|jvr||||jdkrd}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;scorerUg?g)rArRrB)rrLYrws rrwzCheckingClassifier.score7sQ&  E ) )W8M-M-M OOAq ! ! ! >A  EEE rct}d|_d|j_d|j_|S)NTF)super__sklearn_tags__ _skip_test input_tags two_d_array target_tags one_d_labelsrtags __class__s rr{z#CheckingClassifier.__sklearn_tags__Rs9ww''))&+#(,% rNTr)NN) rrrr rrRrTrfrlrtrwr{ __classcell__rs@rr:r:Ds??H# ))))).####J....`???$0BBB66rr:rTF)namekeys validate_keysc<eZdZdZddZdZdZdZfdZxZ S) NoSampleWeightWrapperzWrap estimator which will not expose `sample_weight`. Parameters ---------- est : estimator, default=None The estimator to wrap. Nc||_dSr)est)rrs rrzNoSampleWeightWrapper.__init__ks rc8|j||Sr)rrTrrLrMs rrTzNoSampleWeightWrapper.fitnsx||Aq!!!rc6|j|Sr)rrfrrLs rrfzNoSampleWeightWrapper.predictqsx"""rc6|j|Sr)rrlrs rrlz#NoSampleWeightWrapper.predict_probatsx%%a(((rcVt}d|_|Sr)rzr{r|rs rr{z&NoSampleWeightWrapper.__sklearn_tags__ws$ww'')) rr) rrrr rrTrfrlr{rrs@rrrbs"""###)))rrcfd}|S)Nc(|jduo|jvSrresponse_methods)rmethods rcheckz_check_response..check~s$D0TVt?T5TTrr!)rrs` r_check_responser}s(UUUUU LrceZdZdZd dZdZeeddZeeddZ eed d Z dS) _MockEstimatorOnOffPredictionaEstimator for which we can turn on/off the prediction methods. Parameters ---------- response_methods: list of {"predict", "predict_proba", "decision_function"}, default=None List containing the response implemented by the estimator. When, the response is in the list, it will return the name of the response method when called. Otherwise, an `AttributeError` is raised. It allows to use `getattr` as any conventional estimator. By default, no response methods are mocked. Nc||_dSrr)rrs rrz&_MockEstimatorOnOffPrediction.__init__s 0rc8tj||_|Sr)rJrZr[rs rrTz!_MockEstimatorOnOffPrediction.fits !   rrfcdS)Nrfr!rs rrfz%_MockEstimatorOnOffPrediction.predictsyrrlcdS)Nrlr!rs rrlz+_MockEstimatorOnOffPrediction.predict_probasrrtcdS)Nrtr!rs rrtz/_MockEstimatorOnOffPrediction.decision_functions""rr) rrrr rrTrrrfrlrtr!rrrrs  1111\//),,--.-\///223343\//"56677##87###rr)numpyrJ sklearn.baserr sklearn.utils._metadata_requestsrsklearn.utils.metaestimatorsrsklearn.utils.validationrrr r r r rr:set_fit_requestrrrr!rrrs77777777::::::555555 1 1 1 1 1 1 1 1!B!B!B!B!B!B!B!BHSSSSS-SSSr&3] Ru&&&" M6#####M#####r