`L iEV ddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZmZddlmZddlmZdZd Zd Zd ZdLd Zd ZdZdZdZdZdZ dZ!dZ"ejFjIdgdgdfejJddgddgddgddggejJddgddgddgddggfgdZ&ejFjIdgddZ'dZ(dZ)d Z*d!Z+d"Z,d#Z-d$Z.d%Z/d&Z0d'Z1d(Z2d)Z3d*Z4d+Z5d,Z6d-Z7ejFjId.gd/d0did1fgd/d2ddgid3fejpgd/gd/gd2did3fgd/d2d4id5fejpgd/gd/gd2dd6gid7fggd89d:Z9d;Z:ejFjIdZdLdAZ?dBZ@dCZAdDZBejFjIdgdEdFgdGzfejJddgddgddgddggdFdFggdGzfgdHZCejFjIdgdIdJZDejFjIdgddKZEy)MN)clone)DummyClassifierDummyRegressor)NotFittedError)assert_almost_equalassert_array_almost_equalassert_array_equal)CSC_CONTAINERS)_weighted_percentilec |j|}tj5tjdd|j |}dddt j |}|jdk(rt j|d}|jd}t|}|dk(r|g}g}t|D]}||jd|k(sJ||jdtt j|dd|fk(sJt||jdt jt|tj5tjddtt j |||dddy#1swY^xYw#1swYxYw)Nignorez!divide by zero encountered in log)rraxis) predict_probawarningscatch_warningsfilterwarningspredict_log_probanp atleast_1dndimreshapeshapelenrangeuniquersumoneslog)clfXyproba log_proba n_outputs n_samplesks ^/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/tests/test_dummy.py_check_predict_probar+s   a E  "-*MN))!, - aAvv{ JJq' " IAIA~K 9 FQx~~a I---Qx~~a C !AqD'(:$;;;;!%(,,A,"6AH  $ $ & F  # #H.Q R %bffU1X&6 ! E F F F--( F Fs(F*#A)$A    q! $C 3+ ,, , 3( )) )    1 %C 3+ ,, , 3( )) )r8c dgdgdgdgg}gd}dD]}t|d}|j||t|j|t j t |t||||dk(r9t|j|dg|jjdt|j|dg|jjddkDy)Nrr- most_frequentpriorstrategyrVr[rr?) rr2r r3rr rr+rr class_prior_r)r#r$r]r"s r*%test_most_frequent_and_prior_strategyraas qcA3AA. xa@ 1 3;;q>2773q6?;S!Q' w  %!!1Q4&)3+;+;+C+CG+L  &!!1Q4&)3+;+;+C+CG+Ls+R  r8cdgdgdgdgg}gd}dgdgdgdgg}dD]j}t|d}t|d}|j|||j||t|j||j|ly)Nrr-rr.rYr\rr2r r3)r#y_1dy_2dr]clf_1dclf_2ds r*6test_most_frequent_and_prior_strategy_with_2d_column_yrhus qcA3A D C!qcA3 D.A (C (C 1d 1d6>>!,fnnQ.?@ Ar8c dgdgdgdgg}tjddgddgddgddgg}t|}dD]}t|d}|j ||t |j |tjtj|dftj|dfgt|||t|y)Nrrr.r/)r[rZr\) rr1rrr2r r3hstackr zerosr+r7)r#r$r(r]r"s r*1test_most_frequent_and_prior_strategy_multioutputrls qcA3A 1a&1a&1a&1a&12AAI. xa@ 1  KKN IIrww 1~.)Q0HI J  S!Q'3 r8c:dggdz}gd}td|}|j||dggdz}|j|}tj|t t |z }t|ddd t|d d d t|||y) Nrr:)rr.rrr. stratifiedr\r333333?decimalr.皙? rr2r3rbincountfloatrrr+rTr#r$r"r6ps r*test_stratified_strategyrys  AA <>P QCGGAqM  A [[^F FeCFm+A!gq1!gq1a#r8cdggdz}tjddgddgddgddgddgg}td|}|j||dggdz}|j |}t |j dD]b}tj|dd|ftt|z }t|ddd t|dd d t|||dt|y) Nrr:r.rrnr\rorprqrs rr1rr2r3rrrurvrrr+r7rTr#r$r"r6r)rxs r*$test_stratified_strategy_multioutputr}s  A 1a&1a&1a&1a&1a&9:A <>P QCGGAqM  A [[^F 1771: ( KKq!t %c!f 5AaD'15AaD'15S!Q' ( sr8c:dggdz}gd}td|}|j||dggdz}|j|}tj|t t |z }t|ddd t|d dd t|||y) Nrr;r-uniformr\rorr_rqr.rtrws r*test_uniform_strategyrs  AA 9;M NCGGAqM  A [[^F FeCFm+A!c1-!c1-a#r8cdggdz}tjddgddgddgddgg}td|}|j||dggdz}|j |}t |j dD]b}tj|dd|ftt|z }t|ddd t|ddd t|||dt|y) Nrr;r.rrr\ror_rqr{r|s r*!test_uniform_strategy_multioutputrs  A 1a&1a&1a&1a&12A 9;M NCGGAqM  A [[^F 1771: ( KKq!t %c!f 5AaD#q1AaD#q1S!Q' ( sr8cdggdz}gd}td}|j||t|j|dgdzy)Nrr:)parisrtokyo amsterdamberlinrZr]rrcr#r$r"s r*test_string_labelsrsC  A:A ? 3CGGAqMs{{1~y1}5r8zy,y_test)r.rrr)r.r.rrr.rcntd}|jd||jd|dk(sJy)NrZrr_)rr2score)r$rBr"s r*test_classifier_score_with_Noners5 ? 3CGGD! 99T6 "c )) )r8r])rnrZr[rconstantcgd}dggdz}t||d}|j|||j|}dggdz}t||d}|j|||j|}t||y)Nrr.rrrr;r]rVrr)rr2r3r ) r]rTr$X1clf1 predictions1X2clf2 predictions2s r*+test_classifier_prediction_independent_of_Xrs A #B (:Q D HHRO<<#L #B (:Q D HHRO<<#L|\2r8c"tjj|}dggdz}|jd}t }|j ||t |j|tj|gt|zy)NrGrr;) rrOrPrandnrr2r r3meanrrTrVr#r$regs r*test_mean_strategy_regressorrsq99((.@(AL  A1A  CGGAqMs{{1~ |c!f'<=r8ctjj|}|jdd}|jdd}tj|dj d}|jdd}|jdd}t }|j|||j|}|j|} t||||| t|y)NrG r:rrr^) rrOrPrrrrr2r3rDr7) rTrVX_learnr@rX_testrBr5rArCs r*(test_mean_strategy_multioutput_regressorrs99((.@(AL  R(G  Q'G 777 # + +G 4D   B 'F   A &F  CGGGW;;w'L++f%KdG\6;Osr8ct}tjt5|j gdddy#1swYyxYw)N)rrMraisesrr3)rs r*test_regressor_exceptionsr%s6  C ~ & Bs ?Ac&tjj|}dggdz}|jd}t d}|j ||t |j|tj|gt|zy)NrGrr:medianr) rrOrPrrr2r r3rrrs r*test_median_strategy_regressorr+ss99((.@(AL  A1A ( +CGGAqMs{{1~ ! ~A'>?r8ctjj|}|jdd}|jdd}tj|dj d}|jdd}|jdd}t d }|j|||j|}|j|} t||||| t|y) NrGrr:rrr^rrr) rrOrPrrrrr2r3rDr7) rTrVrr@rrrBr5rArCs r**test_median_strategy_multioutput_regressorr6s99((.@(AL  R(G  Q'G YYwQ ' / / 8F   B 'F   A &F ( +CGGGW;;w'L++f%Kfg|V[Qsr8cHtjj|}dggdz}|jd}t dd}|j ||t |j|tj|gt|zt dd}|j ||t |j|tj|gt|zt dd}|j ||t |j|tj|gt|zt dd}|j ||t |j|tj|d gt|zy) NrGrr:quantiler_r]rrg333333?)q) rrOrPrrr2r r3rrminmax percentilers r* test_quantile_strategy_regressorrKs399((.@(AL  A1A *s ;CGGAqMs{{1~ ! ~A'>? *q 9CGGAqMs{{1~q {SV';< *q 9CGGAqMs{{1~q {SV';< *s ;CGGAqMs{{1~ a2(>'?#a&'HIr8ctjj|}|jdd}|jdd}tj|dj d}tj |ddj d}|jd d}|jd d}td d }|j|||j|} |j|} t||| || t|td d }|j|||j|} |j|} t||| || t|y)NrGrr:rrr^P)rrrrr_rg?) rrOrPrrrrrr2r3rDr7) rTrVrr@rquantile_valuesrrBr5rArCs r*,test_quantile_strategy_multioutput_regressorrbs?99((.@(AL  R(G  Q'G YYwQ ' / / 8FmmG!r:BB7KO   B 'F   A &F *s ;CGGGW;;w'L++f%Kfg|V[Qs *s ;CGGGW;;w'L++f%K, sr8cdggdz}dgdz}tdd}d}tjt|5|j ||dddy#1swYyxYw)Nrr:rrzIWhen using `strategy='quantile', you have to specify the desired quantilematch)rrMr ValueErrorr2r#r$r5err_msgs r*test_quantile_invalidrs`  A aA *t #:BHHiQR^V$WXr8c 6dggdz}|tjddgddgddgddgddgg}td|}tjt d 5|j ||ddddggd z}|j|}t|jdD]f}tj|dd|ftt|z }t|dd d t|dd d t|dd d hy#1swYxYw) Nrr:r.rr;rr\z*the uniform strategy would not save memoryrrogUUUUUU?rq)rr1rrMwarns UserWarningr2r3rrrurvrrrTrr#r$r"r6r)rxs r*+test_uniform_strategy_sparse_target_warningr9s   AbhhAAAAAGHIA 9;M NC k)U V 1   A [[^F 1771: 4 KKq!t %c!f 5AaD%3AaD%3AaD%3 4 s DDc $dggdz}|tjddgddgddgddgddgg}td|}|j||dggdz}|j |}t j |sJ|j}t|jdD]f}tj|dd|ftt|z }t|ddd t|dd d t|dd d hy) Nrr:r;rrnr\rorprqg?)rr1rr2r3rrrrrrurvrrrs r*&test_stratified_strategy_sparse_targetrLs  AbhhAAAAAGHIA <>P QCGGAqM  A [[^F ;;v   ^^ F 1771: 6 KKq!t %c!f 5AaD'15AaD'15AaD'15 6r8c dggdz}|tjddgddgddgddgddgg}t|}tjtj|dftj |dfg}dD]c}t |d}|j|||j|}tj|sJt|j|ey)Nrr:rr/r;rYr\) rr1rrjr rkrr2r3rrr r)rr#r$r( y_expectedr]r"r6s r*3test_most_frequent_and_prior_strategy_sparse_targetr`s  AbhhAAAAAGHIAAIBGGYN3RXXy!n5MNOJ.9xa@ 1 Q{{6"""6>>+Z8 9r8ctjj|}dgg|z}|j|}|j|}t dj |||}|j tj||k(sJt dj |||}|j t||dk(sJt dd j |||}|j t||d k(sJy) NrGrrr)weightsrgI@rgffffff?rgW@) rrOrPrRrr2raverager )rTr(rVr#r$rr5s r*"test_dummy_regressor_sample_weightrps99((.@(AL  A)$A %%i0M & ) - -aM BC ==BJJq-@ @@ @ ( + / /1m DC ==0M4H HH H *t < @ @A} UC ==0M4H HH Hr8ctjdggdggdggg}tjgd}tjgd}t}|j|||j |}t ||y)Nfoobarbazr.r.r.)rr1rr2r3r )r#r$rclsr6s r* test_dummy_regressor_on_3D_arrayrsk E7)ugY% 23A A)$J  CGGAqM [[^Fvz*r8ctjdggdggdggg}gd}gd}dgdgdgg}td}|j|||j |}|j |}t ||t ||y)Nrrrrrrnr)rr1rr2r3rr )r#r$ry_proba_expectedrr6 y_pred_probas r*!test_dummy_classifier_on_3D_arrayrs E7)ugY% 23AAJaS1# < 0CGGAqM [[^F$$Q'Lvz*|%56r8c dggdz}tjgd}tjgd}t}|j|||j |d}t |dk(sJt |d|y) Nrr/r)rrrT) return_stdr.r)rr1rr2r3rr )r#r$y_std_expectedr y_pred_lists r*test_dummy_regressor_return_stdr ss  A AXXi(N  CGGAqM++aD+1K { q  {1~~6r8)rrrr.g?r;cjt}|jd||jd|dk(sJy)Nr)rr2r)r$rBrs r*test_regressor_score_with_Noners3  CGGD! 99T6 "c )) )r8)rrrrcgd}dggdz}t|dd}|j|||j|}dggdz}t|dd}|j|||j|}t||y)Nrrr;gffffff?)r]rrr)rr2r3r )r]r$rreg1rrreg2rs r**test_regressor_prediction_independent_of_Xrs}A #B 8a# FDHHRO<<#L #B 8a# FDHHRO<<#L|\2r8cgd}tjd}t|dd}|j||j |}|j tj k(sJy)Nrr;rr)rrkrr2rdtypefloat64)r]r$r#modelprobass r*test_dtype_of_classifier_probasrsT A  A XA JE YYq!_ * *1 -F <<2:: %% %r8)r)FrnumpyrrM scipy.sparsesparser sklearn.baser sklearn.dummyrrsklearn.exceptionsrsklearn.utils._testingrrr sklearn.utils.fixesr sklearn.utils.statsr r+r7r<rDrWrarhrlryr}rrrmark parametrizer1rrrrrrrrrrrrrrrrrr transposerrrrrrrrrr rrrrr8r*r&s 9- /4F:#"#U *"( A  " $& $$6 |$ BHHq!fq!fq!fq!f5 6 BHHq!fq!fq!fq!f5 6  * * O33&>* @*J.B  4$),'$" $ *,KL  QF+-JK BLL, 5 6 O )   ' = BLL, 5 6 !Y ( B & +:;:H.9 : .94:4$.96:6&.9 9: 9I"+ 7 7 vz" Aq6Aq6Aq6Aq62 3tTl^a5GH** %OP 3Q 3O&&r8