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%j/home/kim/smarthome/.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$predZprobaZ st_stringZ pred_stringZ proba_stringZlabeledr%r%r&test_classification.s0        &&r?cCsttdddddd}tt}d|dd<tjd}t|dd}|t||j |ks0Jt |j dkdksZ max_probaZmost_confident_svcZ added_by_strowr%r%r&test_k_best_selects_bests  rqcCstdtddfdtddfgtdddd}t|dsJt|d}|tt|ttdtd dfdtd dfgtd ddd}t|drJJt|d}t t |ttWddS1sfwYdS) NZsvc_1T)r,Zsvc_2r)Z estimatorsZfinal_estimatorZcvr3r(F) rr hasattrr rr r!r3r2rrRAttributeError)r(rar%r%r&test_estimator_meta_estimators.         "rucCstddd}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'rNrrzBThis 'SelfTrainingClassifier' has no attribute 'decision_function'zD'DecisionTreeClassifier' object has no attribute 'decision_function')r r rrRrtrr r!rdecision_function isinstancevalue __cause__str)r(Z self_trainingZ outer_msgZ inner_msgZ exec_infor%r%r&,test_self_training_estimator_attribute_error@s  r{cCsd}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.)r|r() rr FutureWarningr rrr r!rRr`)Zwarn_msg error_msgr%r%r&'test_deprecation_warning_base_estimator^s"  "rz2ignore:y contains no unlabeled samples:UserWarningmethod)rvZpredict_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`.rrz1is only supported if enable_metadata_routing=Truerra)Z sample_weightpropNT)r r rrRr`rZfitted_params_getattr)rZestr%r%r&*test_routing_passed_metadata_not_supportedts " "r)Bmathrnumpyr5rZ numpy.testingrZsklearn.datasetsrrZsklearn.ensemblerZsklearn.exceptionsrZsklearn.metricsrZsklearn.model_selectionr Zsklearn.neighborsr Zsklearn.semi_supervisedr Z sklearn.svmr Zsklearn.tests.test_pipeliner Z sklearn.treerZirisdatatargetr r2rBrLrKrAr!r7r6r8Zastypeobjectr4r'markZ parametrizer?rIrMrNrQrTrDrWrYr[rcrjrmrqrur{rfilterwarningsrr%r%r%r&sx               )       $