sg\8 dddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZeZeej:ej<d\ZZ Z!Z"dZ#e!jIZ%de%e#ddddddZ&ejNe&jPe%jSe*Z+de+e%dk(<dZ,ejZj]deedddgejZj]dddgdZ/dZ0d Z1d!Z2ejZj]deedddgejZj]d"e%e+gd#Z3d$Z4ejZj]d%e5d&d'd(Z6d)Z7d*Z8d+Z9ejZj]d,dd-gd.Z:d/Z;d0Zd3Z?ejZjd4ejZj]d5gd6d7ZAy)8)ceilN)assert_array_equal) load_iris make_blobs)StackingClassifier)NotFittedError)accuracy_score)train_test_split)KNeighborsClassifier)SelfTrainingClassifier)SVC)SimpleEstimator)DecisionTreeClassifier) random_state2ABCz-1)rrcttdd}tjtd5|j t tddd|jdk(sJy#1swYxYw)Nk_besti) criterionrzk_best is larger thanmatch all_labeled) r r pytestwarns UserWarningfitX_trainy_train_missing_labelstermination_condition_sts c/var/www/html/venv/lib/python3.12/site-packages/sklearn/semi_supervised/tests/test_self_training.pytest_warns_k_bestr(&s^  4 6(SW XB k)@ A0 w./0  $ $ 55 500s A''A0 estimatorscaleTgamma probabilityrselection_crit thresholdrc(d}d}t||||}|jtt|j t }|j t }t||||}|jtt|j t }|j t } ttjtj||t|| |j|jk(sJtdk7} t|jdk(| tt| |j| tj |j|j"cxkr|ksJJtj |j|j"cxkr|ksJJ|jj$|jj$k(sJ|jj$|jj$k(sJy)Ng? )max_iterr/r)r2rr/rr)r r!r"r#predictX_test predict_probay_train_missing_stringsrnp vectorizemappinggetr$ labeled_iter_ transduction_maxn_iter_shape) r)r.r/r2r&predproba st_string pred_string proba_stringlabeleds r'test_classificationrF.sIH H ^ BFF7*+ ::f D   V $E&H)IMM'23##F+K**62L0r||GKK06 Dul+ $ $ (H(H HH H$*Gr''1,g6-g68H8H8QR 66""" #rzz =X == == = 66))) *i.?.? K8 KK KK K    ! !R%5%5%;%; ;; ;  " " ( (I,C,C,I,I II IcBttdddd}tjt}d|ddtj d}t |dz dz }|jt||j|k(sJtj|jdk(dk(sJtd|D])}tj|j|k(dk(r)Jtj|j|k(|dz dzk(sJ|jdk(sJy) Nr n_neighborsrr1)rrr2rrr)r r r7copyy_trainr?rr!r"r>sumr;ranger$)r&y_train_only_one_label n_samplesn_expected_iteris r' test_k_bestrS\s +  B  WWW-!#12 a IIMR/0OFF7*+ :: (( ( 66"""a' (A -- - 1o &3vvb&&!+,2223 66"""o5 69q=B:N NN N $ $ 55 5rGctdd}|jttdttdt |}|jtt |jt|jt}}tj||rJt|jtt}t|jtt}||kDsJy)Nr*Tr,r-) r r!r"n_labeled_samplesrLr r#r3r4r7 array_equalr y_test)r)r&pred1pred2score_supervisedscore_self_trainings r'test_sanity_classificationr]ss't4I MM'+,-w7H7I/JK  *BFF7*+$$V,bjj.@5E~~eU++ +%i&7&7&?H(F);VD !1 11 1rGcttdd}|jtt|j dksJ|j dk(sJy)Ng?)r/r2r1r)r r r!r"r#r>r$r%s r'test_none_iterr_sJ 4 6$QU VBFF7*+ ::?? $ $ 55 5rGyct|d}|jt||jtdt|dt}t |j t |j t |jdk(sJy)Nrr2r2)r r!r"rVrr3r4r$)r)r`clf1clf2s r'test_zero_iterationsresp ")a 8DHHWa ==!3"34a8J9J6K LDt||F+T\\&-AB  & &* 44 4rGct}|jttt |}t j td5|jtdddy#1swYyxYw)Nz6This SelfTrainingClassifier instance is not fitted yetr) r r!r"rLr rraisesrr3knnr&s r'test_prefitted_throws_errorrjsY  CGGGW  $B F  7 s A**A3r2rc0tt|}|jttt |j |j dk(}|tk(sJtj|j |jcxkr|ksJJy)Nrbr) r r r!r"r#lenr;rVr7r=r>)r2r& amount_iter_0s r'test_labeled_iterros 4 6 JBFF7*+(()9)9Q)>?@M - -- - 66""" #rzz =X == == =rGct}|jttt |}t j td5|jttdddt|jt|jttj|jdk(sJ|jdk(sJy#1swYqxYw)Nzy contains no unlabeled samplesrrr)r r!r"rLr rrr rr3r4r7allr;r$rhs r'test_no_unlabeledrrs  CGGGW  $B k)J K! w !s{{6*BJJv,>? 66"""a' (( ( $ $ 55 5 !!s CCctdd}t|}dgdgdgdgg}gd}|j|||jdk(sJ|jdk(sJy) Nr*TrUrrg?)rrrr no_change)r r r!r>r$)svcr& X_train_easy y_train_easys r'test_early_stoppingrxsg G .C  $BC!qcC5)L!LFF<& ::?? $ $ 33 3rGctt}tddd\}}gd}tj||}t j td5|j||dddy#1swYyxYw)Nrg?)rPr cluster_std)onetwothreedtyper) r r rr7takerrg ValueErrorr!)clfXr`labels_multiclass y_stringss r'test_strings_dtypersj !5!7 8C  DDAq/)1-I z 1 9s A66A?verboseFctt|}|jtt|j }|rd|j vsJyd|j vsJy)N)r iteration)r r r!r"r# readouterrout)capsysrrcaptureds r' test_verbosersV !5!7 ICGGG+,  "Hhll***(,,...rGcttddddd}tjt}d|ddtj d}t |dz dz }|jt||j}d }td|D]"}|j|d|jvr"J|j||dz dz|jvsJy) NrrIrr1T)rrrr2rrz)End of iteration {}, added {} new labels.) r r r7rKrLr?rr!r"rrNformatr)rr&rOrPrQrmsgrRs r'test_verbose_k_bestrs +  B WWW-!#12 a IIMR/0OFF7*+  "H 5C 1o &1zz!R HLL0001 ::o A '; < LL LrGctddd}t|ddd}td k7}|jtt||j d k7z}|jt|t||j t|}tj|d }t|tj|d d}ttj|j}|jD]}||vrJy) Nr*Trr+rrr1)rr2rr)axisi) r r r#r!r"r<r5r7r=argsortwheretolist) rur& has_label got_labelr@ max_probamost_confident_svc added_by_strows r'test_k_best_selects_bestrs GA >C x!B OB&",IFF7*+ b.."45IGGGI  6y AB   WiZ0 1Dt!$I ),RZZ -B34-HI"((9-.557K!((*"k!!!"rGc"tdtdfdtdfgtdd}t|dsJt|}|j t t |jttdtd fdtd fgtd d}t|drJt|}tjt5|j t t dddy#1swYyxYw) Nsvc_1T)r-svc_2r) estimatorsfinal_estimatorcvr5r)F) rr hasattrr r!r"r#r5r4rrgAttributeError)r)rs r'test_estimator_meta_estimatorrs# cd+ , cd+ , - I 9o .. . 9 5CGGG+,f" ce, - ce, - . Iy/22 2 9 5C ~ &1 /0111s !DDcDtdd}t|}tjtd5|j t tdddtt}d}d }tjt|5}|j t tjt dddtjjtsJ|t|jjvsJy#1swYxYw#1swY_xYw) 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 rrgrr!r"r#rdecision_function isinstancevalue __cause__str)r) self_training outer_msg inner_msg exec_infos r',test_self_training_estimator_attribute_errorr@sW5I*95M ~-O P;'#9:; +5K5MNMTIVI ~Y 7V9'#9:LLWUV ioo// @@ @ IOO556 66 6;;VVsD  .D DDc6d}tjt|5tt j t tdddd}tjt|5tj t tdddd}tjt|5tt t j t tdddy#1swYxYw#1swYqxYw#1swYyxYw)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 rr!r"r#rgr)warn_msg error_msgs r''test_deprecation_warning_base_estimatorr^sPH m8 4 .D.FGKK +  GI z 3F $$W.DEFNI z 3/13?U?W #g- .//  FF//s#-C70#D86D7DD Dz2ignore:y contains no unlabeled samples:UserWarningmethod)rpredict_log_probar5r3ctt}tjtd5|j dgdggddgdgddddtt}tjtd5d|_t||dggdgddddy#1swYaxYw#1swYyxYw) 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 rrrgrr!fitted_params_getattr)rests r'*test_routing_passed_metadata_not_supportedrts !?+< =C M A !qc QF1#C@A !?+< =C M A"VqcU1#C@ AA AA AAsB- B9-B69C)Bmathrnumpyr7r numpy.testingrsklearn.datasetsrrsklearn.ensemblersklearn.exceptionsrsklearn.metricsr sklearn.model_selectionr sklearn.neighborsr sklearn.semi_supervisedr sklearn.svmr sklearn.tests.test_pipeliner sklearn.treeririsdatatargetr"r4rLrXrVrKr#r9r8r:astypeobjectr6r(mark parametrizerFrSr]r_rerjrNrorrrxrrrrrrrfilterwarningsrrGr'rs` ,2/-*42:7/ {#3IIt{{$ & -/()*cct ,3,",,w{{34JKRR 9;."456SwDqQR)K+BC&JD  &JR6. 26SwDqQR57NOP 5Q  5 U1a[1 >2 > 6 4T5M2 /3 /M2"(!1H7</,PQ TARArG