rtjOddlZddlmZddlmZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZmZddlmZddlmZmZdd lmZdd lmZmZmZdd lmZd6d Z e!fdZ"ee dZ#d7dZ$dZ%Gdde&Z'Gdde e Z(Gdde e Z)Gdde e Z*Gdde e Z+Gdde+Z,Gdd e+Z-Gd!d"e+Z.Gd#d$ee Z/Gd%d&e Z0d7d'Z1Gd(d)eZ2Gd*d+eeZ3Gd,d-e3eZ4Gd.d/e e e Z5Gd0d1e e e Z6Gd2d3e e e Z7Gd4d5e ee Z8dS)8N) defaultdict)partial)assert_array_equal) BaseEstimatorClassifierMixinMetaEstimatorMixinRegressorMixinTransformerMixinclone)_Scorermean_squared_error)BaseCrossValidator) GroupKFoldGroupsConsumerMixin)SIMPLE_METHODS)MetadataRouter MethodMappingprocess_routing)_check_partial_fit_first_callTc 6tj}|dj}|dj}t|dst d|_|sd|D}|j|||dS)zUtility function to store passed metadata to a method of obj. If record_default is False, kwargs whose values are "default" are skipped. This is so that checks on keyword arguments whose default was not changed are skipped. _recordsc*ttSN)rlistc/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sklearn/tests/metadata_routing_common.pyz!record_metadata..*s;t+<+<rcPi|]#\}}t|tr|dk ||$S)default isinstancestr).0keyvals r z#record_metadata..,sG   Sc3'' -09,<,< ,<,<,"#<#<==    "LLNN    L ''/////rc t|dt|t|t}|D]}t |t |ks4Jd|d||D]\}}||} ||vr,| *tj| | sJ=t| tj rt| |h| |usJd| d|d|dS)aCheck whether the expected metadata is passed to the object's method. Parameters ---------- obj : estimator object sub-estimator to check routed params for method : str sub-estimator's method where metadata is routed to, or otherwise in the context of metadata routing referred to as 'callee' parent : str the parent method which should have called `method`, or otherwise in the context of metadata routing referred to as 'caller' split_params : tuple, default=empty specifies any parameters which are to be checked as being a subset of the original values **kwargs : dict passed metadata rz Expected z vs Nz . Method: ) getattrdictgetrsetkeysr.npisinallr$ndarrayr) r0methodparent split_paramsr2 all_recordsrecordr'valuerecorded_values rcheck_recorded_metadatarG4s( Z((,,VTVV<<@@PP6;;==!!S%7%7777 : : :6;;== : :877!,,..  JC#C[Nl""~'Aw~u5599;;;;;;nbj99&~u====)U222QNQQQQQQ3222  rF)r1c*t|tr1|D],\}}| ||vr ||}nd}t|j|-dS|gn|}tD]>}||vrt ||}d|jD}|rJ?dS)aCheck if a metadata request dict is empty. One can exclude a method or a list of methods from the check using the ``exclude`` parameter. If metadata_request is a MetadataRouter, then ``exclude`` can be of the form ``{"object" : [method, ...]}``. N)excludecFg|]\}}t|ts||Srr#)r&propaliass r z+assert_request_is_empty..ws@   e%%% */): ):):):r)r$rassert_request_is_emptyrouterrr7requestsr.)metadata_requestrIname route_mapping_excluder@mmrpropss rrNrNbs"N33#3 L L D-"tw"4= #M$8( K K K K KObbG   W   &//  "|1133    y  rcD]"\}}t||}|j|ksJ#fdtD}|D]&}t t||jrJ'dS)Ncg|]}|v| Srr)r&r@ dictionarys rrMz(assert_request_equal..s#UUUF* NNXNNNrrc(eZdZdZedZdS))ConsumingClassifierWithoutPredictLogProbazConsumingClassifier without a predict_log_proba method, but with predict_proba. Used to mimic dynamic method selection such as in `BaggingClassifier.predict_log_proba()`. c tdNz1This estimator does not support predict_log_probarrfs rrz;ConsumingClassifierWithoutPredictLogProba.predict_log_proba[PQQQrN)rhrirjrrrrrrrrTs> RRXRRRrrc>eZdZdZedZedZdS)"ConsumingClassifierWithOnlyPredictzConsumingClassifier with only a predict method. Used to mimic dynamic method selection such as in `BaggingClassifier.predict_log_proba()`. c tdrrrfs rrz0ConsumingClassifierWithOnlyPredict.predict_probagrrc tdrrrfs rrz4ConsumingClassifierWithOnlyPredict.predict_log_probakrrN)rhrirjrrrrrrrrr`s\ NNXNRRXRRRrrc:eZdZdZd dZd dZd dZd dZd dZdS) ConsumingTransformera~A transformer which accepts metadata on fit and transform. Parameters ---------- registry : list, default=None If a list, the estimator will append itself to the list in order to have a reference to the estimator later on. Since that reference is not required in all tests, registration can be skipped by leaving this value as None. Nc||_dSrrnrps rrqzConsumingTransformer.__init__|rrrr"cz|j|j|t|||d|_|S)NruT)ror/ryfitted_rzs rrzConsumingTransformer.fitsL = $ M  & & &#        rc0t||||dzSrrrs r transformzConsumingTransformer.transform+#      1u rct|||||||||||Srt)ryrrrzs r fit_transformz"ConsumingTransformer.fit_transforms] $      xx1MHxMMWW ]XX   rc0t||||dz Srrrs rinverse_transformz&ConsumingTransformer.inverse_transformrrrrrNN) rhrirjrrqrrrrrrrrrps  !!!!     rrc*eZdZdZddZddZddZdS) "ConsumingNoFitTransformTransformerzA metadata consuming transformer that doesn't inherit from TransformerMixin, and thus doesn't implement `fit_transform`. Note that TransformerMixin's `fit_transform` doesn't route metadata to `transform`.Nc||_dSrrnrps rrqz+ConsumingNoFitTransformTransformer.__init__rrrcl|j|j|t||||Srt)ror/r5rzs rrz&ConsumingNoFitTransformTransformer.fits: = $ M  & & &MHMMMM rc*t||||Srt)r5rs rrz,ConsumingNoFitTransformTransformer.transformsMHMMMMrrNNNr)rhrirjrrqrrrrrrrs\QQ!!!!rrc ||tttfi||dd}t |||S)Nrvrv)r/consuming_metricryr9r )ry_trueror2rvs rrrsZ())) 0;;F;;;JJ55M ffM J J JJrc eZdZdfd ZxZS)ConsumingScorerNctt|}t|did||_dS)Nrnrr) score_funcsignr2response_method)rrsuperrqro)raror __class__s rrqzConsumingScorer.__init__sM-AAA  !"i    ! rr)rhrirjrq __classcell__)rs@rrrs=!!!!!!!!!!rrc.eZdZddZddZd dZd dZdS) ConsumingSplitterNc||_dSrrnrps rrqzConsumingSplitter.__init__rrrr"c#:K|j|j|t|||t|dz}t t d|}t t |t|}||fV||fVdS)N)groupsrwrr)ror/ryrZrrange)rar{r|rrw split_index train_indices test_indicess rsplitzConsumingSplitter.splits = $ M  & & &#D(KKKK!ffk U1k2233 E+s1vv6677 M))))\))))))rcdS)Nrr)rar{r|rrws r get_n_splitszConsumingSplitter.get_n_splitssqrc#Kt|dz}ttd|}tt|t|}|V|VdS)Nrr)rZrr)rar{r|rrrrs r_iter_test_indicesz$ConsumingSplitter._iter_test_indicessg!ffk U1k2233 E+s1vv6677 rrr)NNNNr)rhrirjrqrrrrrrrrsd!!!! * * * *rrceZdZdZdS))ConsumingSplitterInheritingFromGroupKFoldz\Helper class that can be used to test TargetEncoder, that only takes specific splitters.N)rhrirjrrrrrrsrrc$eZdZdZdZdZdZdS) MetaRegressorz(A meta-regressor which is only a router.c||_dSr) estimator)rars rrqzMetaRegressor.__init__s "rc t|dfi|}t|jj||fi|jj|_dSNr)rr rr estimator_rar{r| fit_paramsparamss rrzMetaRegressor.fitsK u;; ;;3%//3AqQQFsN u;; ;;7E$"23371WW@R@VWW rc Xt|dfi|}|jj|fi|jjS)Nr)rr1rr/)rar{r|transform_paramsr s rrzMetaTransformer.transformCs> {GG6FGG*t *1MM0B0LMMMrct||jtddddS)Nrrrr)r/r)rrr/rrfs rrz$MetaTransformer.get_metadata_routingGsYD)))--((?? SeS , , S KS 8 8 .   rr)rhrirjrrqrrrrrrr-r-8sd$$''' NNNN     rr-)Tr)9r* collectionsr functoolsrnumpyr< numpy.testingr sklearn.baserrrr r r sklearn.metrics._scorerr r sklearn.model_selectionrsklearn.model_selection._splitrr sklearn.utils._metadata_requestsrsklearn.utils.metadata_routingrrrsklearn.utils.multiclassrr5tuplerGryrNr]rr_rlrrrrrrrrrrrrrrr(r-rrrrAs######,,,,,,@???????666666JJJJJJJJ CBBBBB0000,?Degg((((V&goeLLL::::        ++++++++\ % % % % %_m % % %F     NM   QQQQQ/=QQQh N N N N N-@ N N N R R R R R0C R R R R R R R R)< R R R /////+]///d*KKKK!!!!!g!!!+-?60A: & $. D/-8     (*:M     r