`L i#&ddlZddlZddlmZmZmZddlmZm Z m Z m Z ddl m Z mZmZmZmZmZddlmZddlmZmZddlmZddlmZmZmZmZe d \ZZ ed \Z!Z"ejFjId ge d edefdefde ddfgdge d edefdefde ddfgge d edefdefdeddfgdge d edefdefdeddfgggddZ%ejFjIdeeegdZ&ejFjIdge d ege d ege d ege d egdZ'ejFjId ge d edefgge d edefgge d edefgge d edefgggddZ(ejFjIdeeee feee!e"feeee feee!e"fgd Z)y)!N)ClassifierMixinclone is_classifier) load_diabetes load_irismake_classificationmake_regression)RandomForestClassifierRandomForestRegressorStackingClassifierStackingRegressorVotingClassifierVotingRegressor) SimpleImputer)LinearRegressionLogisticRegression) make_pipeline)SVCSVR LinearSVC LinearSVRT) return_X_yzX, y, estimator  n_sampleslrsvmrf) n_estimators max_depth) estimatorscvr$)zstacking-classifierzvoting-classifierzstacking-regressorzvoting-regressor)idscd|jvsJ|jj|jddusJ|jj|jdusJ|j||t |jdk(sJt |j dk(sJt t|j jt gdk(sJt|}t|r tn t}|j|j||t|drJ|jjj!|jjj!k(sJ|jj"j!|jj"j!k(sJt|}|jd|j||t |jdk(sJ|jjdk(sJt |j dk(sJt t|j jt gdk(sJ|j D]-}t%|t'|jjs-J|jd|jd |j!d |j!dj!d k(sJ|j!d |j!dj!dk(sJy)Nrr )rrr)rdropg$@)svm__Cr) rf__max_depthr+Cr,rr")named_estimatorsrr$fitlennamed_estimators_sortedlistkeysrrrr set_paramshasattrr get_paramsr isinstancetype)Xy estimatorestimator_new_params svm_estimatorestimator_droppedsub_ests h/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/ensemble/tests/test_common.py/test_ensemble_heterogeneous_estimators_behaviorrBsAv I.. .. .  % % ) )Y-A-A!-DQ-G GG G  % % ) )Y-G-G-N NN NMM!Q y)) *a // / y** +q 00 0 $y22779: ;v@   !+*95CE35M## #6::1a@+U33 3--00;;=  % % ( ( 3 3 5 6 6 --00;;=  % % ( ( 3 3 5 6 6 i(  V ,!Q  11 2a 77 7  - - 1 1V ;; ;  22 3q 88 8 $(::??AB CvH  %66MgtI,F,F,J,J'KLLLM % q)x(    !% ( 3 3 5c : ; ; /    !$ ' 2 2 4[ A B BEnsemblecVt|trtd\}}dtfg}d}nt d\}}dt fg}d}||}dj |}tjt|5|j||dddy#1swYyxYw) Nrrr classifier regressorr&zshould be a {}match) issubclassrrrr rformatpytestraises ValueErrorr/)rDr:r;r$ ensemble_typeensembleerr_msgs rA+test_ensemble_heterogeneous_estimators_typerRs(O,"R01-/01 $ ,1/123 # :.H%%m4G z 1 Qs BB(zX, y, Ensemblect|trdtfg}n dtfg}||}d}t j t |5|j||dddt|trdtfdtfg}ndtfdtfg}||}d}t j t |5|j||dddt|trdtfg}n dtfg}||}d}t j t |5|j||dddy#1swYxYw#1swYxYw#1swYyxYw) Nlr__r&z3Estimator names must not contain __: got \['lr__'\]rHrz-Names provided are not unique: \['lr', 'lr'\]r$z3Estimator names conflict with constructor arguments)rJrrrrLrMrNr/)r:r;rDr$rPrQs rA6test_ensemble_heterogeneous_estimators_name_validationrUsg(O,1345 /123 :.HDG z 1 Q(O,/12T;M;O4PQ -/049I9K2LM :.H>G z 1 Q(O,#%7%9:; #%5%789 :.HCG z 1 Q-s$EE(E4E%(E14E=c|jdtjtd5|j ||dddy#1swYyxYw)Nr*)rzAll estimators are dropped.rH)r5rLrMrNr/)r:r;r<s rA2test_ensemble_heterogeneous_estimators_all_droppedrWsG<F# z)F G as A  AzEnsemble, Estimator, X, ycd|j}tjjddg|jddgj t }tj||<tt|}|d|fd|fg}|j||j||y) Nr)rg?g?)ppipe1pipe2r&) copynprandomchoiceshapeastypeboolnanrrr/score)rD Estimatorr:r;maskpiperPs rA2test_heterogeneous_ensemble_support_missing_valuesrhs A 99  QFAGGSz  : A A$ GDffAdG )+ 6DWdOgt_#EFH LLAQ"rC)*numpyr]rL sklearn.baserrrsklearn.datasetsrrrr sklearn.ensembler r r r rrsklearn.imputersklearn.linear_modelrrsklearn.pipeliner sklearn.svmrrrrr:r;X_ry_rmark parametrizerBrRrUrWrhrCrArvs >> )E*66D!1 D )S 2 . -/0IK(1qANO     2 . -/0IK(1qANO   r * +-.IK(0a1MN     r * +-.IK(0a1MN  C+X ]4j8k4j8v(/: &@  +@-?@>  +>-=>;/B ';):;9/B '99   F 2 . D2D2F+G*H I  2 . $0B0D)E(F G  r * 41A1C*D)E F  r * /?/A(B'C D $ )676 /A6 ,c37 -q!4 *C5 ##rC