o \i\8 @sDddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZeZeejejdd\ZZ Z!Z"dZ#e!$Z%de%e#d<dddddZ&e'e&j(e%)e*Z+de+e%dk<ddZ,ej-.deeddddgej-.dddgdd Z/d!d"Z0d#d$Z1d%d&Z2ej-.deeddddgej-.d'e%e+gd(d)Z3d*d+Z4ej-.d,e5d-d.d/d0Z6d1d2Z7d3d4Z8d5d6Z9ej-.d7dd8gd9d:Z:d;d<Z;d=d>ZdCdDZ?ej-@dEej-.dFgdGdHdIZAdS)J)ceilN)assert_array_equal) load_iris make_blobs)StackingClassifier)NotFittedError)accuracy_score)train_test_split)KNeighborsClassifier)SelfTrainingClassifier)SVC)SimpleEstimator)DecisionTreeClassifier) random_state2ABCz-1)rrcCs\ttddd}tjtdd|ttWdn1s wY|jdks,JdS)Nk_besti) criterionrzk_best is larger thanmatch all_labeled) r r pytestwarns UserWarningfitX_trainy_train_missing_labelstermination_condition_str%/var/www/www-root/data/www/176.119.141.140/sports-predictor/venv/lib/python3.10/site-packages/sklearn/semi_supervised/tests/test_self_training.pytest_warns_k_best&s r' estimatorscaleTgamma probabilityrselection_crit thresholdrc Cs<d}d}t||||d}|tt|t}|t}t||||d}|tt|t}|t} tt t j ||t|| |j |j ksLJtdk} t|jdk| tt| |j| t |j|jkrr|ksuJJt |j|jkr|ksJJ|jj|jjksJ|jj|jjksJdS)Ng? )max_iterr.r)r0rr.rr)r rr r!predictX_test predict_probay_train_missing_stringsrnp vectorizemappinggetr" labeled_iter_ transduction_maxn_iter_shape) r(r-r.r0r$predproba st_string pred_string proba_stringlabeledr%r%r&test_classification.s0        &&rDcCsttdddddd}tt}d|dd<tjd}t|dd}|t||j |ks0Jt |j dkdks max_probamost_confident_svc added_by_strowr%r%r&test_k_best_selects_bests  rcCstdtddfdtddfgtdddd}t|dsJt|d}|tt|ttdtd dfdtd dfgtd ddd}t|drJJt|d}t t |ttWddS1sfwYdS) Nsvc_1T)r,svc_2r) estimatorsfinal_estimatorcvr3r(F) rr hasattrr rr r!r3r2rr_AttributeError)r(rvr%r%r&test_estimator_meta_estimators.         "rcCstddd}t|}tjtdd|ttWdn1s"wYttd}d}d }tjt|d}|tt tWdn1sLwYt |j j tsZJ|t |j j vsdJdS) a>Check that we raise the proper AttributeErrors when the `estimator` does not implement the `predict_proba` method, which is called from within `fit`, or `decision_function`, which is decorated with `available_if`. Non-regression test for: https://github.com/scikit-learn/scikit-learn/issues/28108 Fr))r,r+z has no attribute 'predict_proba'rNrzBThis 'SelfTrainingClassifier' has no attribute 'decision_function'zD'DecisionTreeClassifier' object has no attribute 'decision_function')r r rr_rrr r!rdecision_function isinstancevalue __cause__str)r( self_training outer_msg inner_msg exec_infor%r%r&,test_self_training_estimator_attribute_error@s  rcCsd}tjt|dttdttWdn1swYd}tjt |dtttWdn1s>wYd}tjt |dtttdttWddS1sdwYdS)Nz?`base_estimator` has been deprecated in 1.6 and will be removedr)base_estimatorz4You must pass an estimator to SelfTrainingClassifierz;You must pass only one estimator to SelfTrainingClassifier.)rr() rr FutureWarningr rrr r!r_ru)warn_msg error_msgr%r%r&'test_deprecation_warning_base_estimator^s"  "rz2ignore:y contains no unlabeled samples:UserWarningmethod)rpredict_log_probar3r1cCsttd}tjtdd|jdgdggddgdgddWdn1s(wYttd}tjtddd|_t||dggdgddWddS1sVwYdS) zTest that the right error message is raised when metadata is passed while not supported when `enable_metadata_routing=False`.rz1is only supported if enable_metadata_routing=Truerra) sample_weightpropNT)r r rr_rurfitted_params_getattr)restr%r%r&*test_routing_passed_metadata_not_supportedts " "r)Bmathrnumpyr5r numpy.testingrsklearn.datasetsrrsklearn.ensemblersklearn.exceptionsrsklearn.metricsrsklearn.model_selectionr sklearn.neighborsr sklearn.semi_supervisedr sklearn.svmr sklearn.tests.test_pipeliner sklearn.treeririsdatatargetr r2rHrSrQrGr!r7r6r8astypeobjectr4r'mark parametrizerDrOrXrYr^rbrJrfrhrmrzrrrrrrfilterwarningsrr%r%r%r&sx               )       $