iLpddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z ddl mZmZmZddlmZddlmZmZmZddlmZmZmZmZd Zd Zej<j?d dgezej<j?d d dgdZ dZ!ej<j?dddgdZ"dZ#dZ$ej<j?d d dgej<j?dedZ%dZ&ej<j?dedZ'dZ(dZ)dZ*dZ+ej<j?d dgezej<j?d d dgd!Z,d"Z-d#Z.ej<j?d dgezd$Z/ej<j?d%dd gej<j?d dgezd&Z0ej<j?d'ed(Z1ej<j?ded)Z2ej<j?d dgezej<j?d*d+d,Z3ej<j?d%dd gej<j?d dd gd-Z4ej<j?d.dd/gej<j?d dgezd0Z5ej<j?ded1Z6ej<j?d2gd3ej<j?d dgeDcgc]0}ejn|ej<jqd4562c}zej<j?d dd gd7Z9ycc}w)8N)linalgsparse) load_irismake_regressionmake_sparse_uncorrelated)LinearRegression)_preprocess_data _rescale_data make_dataset)add_dummy_feature)assert_allcloseassert_array_almost_equalassert_array_equal)COO_CONTAINERSCSC_CONTAINERSCSR_CONTAINERSLIL_CONTAINERSư>cdgdgg}ddg}t}|j||t|jdgt|jdgt|j |ddgdgg}dg}t}|j||t|jdgt|jdgt|j |dgy)Nr)rfitrcoef_ intercept_predict)XYregs /var/www/ITT-ServiceAPI-Services/ITT-Document-Scan/mrzscanner/mrzenv/lib/python3.12/site-packages/sklearn/linear_model/tests/test_base.pytest_linear_regressionr !s qc A AA  CGGAqMcii!-cnnqc2ckk!nq!f5 A A  CGGAqMcii!-cnnqc2ckk!nqc2sparse_container fit_interceptTFctjj|}d\}}|j||f}|||}|j|}d|j |z}t |d} | j |||| j} | j} | jj|jdfk(sJtj|} |s|n t|} | j| z| z}| j| z|z}tj||}|s t| |yt| |ddt| |dy) N))size?gؗҜ<)r#tol sample_weightrr)nprandom RandomStatenormaluniformrrrrshapediagr Trsolver )r"r#global_random_seedrng n_samples n_featuresryr+rcoefs1inter1WX_augXwywcoefs2s r%test_linear_regression_sample_weightsrA9s> ))   2 3C!Iz J/ 0A# Q  "A#++9+55M E BCGGAq G. YYF ^^F 99??qwwqzm ++ +  A"A(9!(?ckk!nqwwy8!O>O>Q6RR'(]## Q /?  #' 0,BFFGW# IIK ZZ\fo6fo6grwwz':;!78  ' B4 01 4 5 B4 0A4G H IGq?':;<  ' B " BO 3 4A/0r! lil_containercPtjj|}d}d}tj||d|}||}|j |}|j }t ||d\}} } } } } t| tj|t| dt| tj|| Jt|j |t| |t ||d\}} } } } } t| tj|d t| tj|d t| tj|| Jt|j |t| |tj|d z y) Nrkrr)densityrwFrWrTr) r,r-r.rrcrr rrrIr)r5rr6r7r8rr9XArrrrrrs r#test_sparse_preprocess_data_offsetsrs] ))   2 3CIJ Iz3SIAaA A B/? 1E0,BFFGWfbhhz&:;fa(grwwz':; ??bjjlB/b!$/? 1D0,BFFGWfbggbq&9:fbggaa&89grwwz':; ??bjjlB/b!bggaa&8"89r!ct\}}d||dk<||}t||d\}}}}}}|jdk(sJy)Nrl@TrWcsr)rr format)rirr9rcsr_rs rtest_csr_preprocess_datarsT  DAqAa#gJ  C*3FD!Q1a ;;%  r!to_copy)TFct\}}d||dk<|||}t||d|d\}}}}}}|r/|-tj|j|jrJy|rtj||rJy|-tj|j|jsJytj||sJy)NrlrTF)r#r check_input)rr r,may_share_memorydata)r"rrr9X_y_rs r#test_preprocess_copy_data_no_checksrs  DAqAa#gJ# Q ) 1DwEBAq!Q#/&&rww7777 &&r1----  %""277AFF333""2q)))r!c tjj|}d}d}|j||}|j|}|j|dz}tj|tj } tj|tj } tj|tj } tj|tj } tj|tj } tj|tj }t| | || |\}}}}}}t| | |||\}}}}}}t| | || |\}}}}}} t| | |||\}!}"}#}$}%}&|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|r|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|r|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|r| jtj k(sJ|!jtj k(sJ|"jtj k(sJ|#jtj k(sJ|$jtj k(sJ|jtj k(sJ|r|&jtj k(sJ| jtj k(sJ| jtj k(sJ| jtj k(sJ| jtj k(sJt||ddt||ddt||d t||d t|||rt||d yy) Nrkrrrrrr)rtolatolr) r,r-r.rcasarrayfloat32float64r rr )'rr#r5r6r7r8rr9swX_32y_32sw_32X_64y_64sw_64Xt_32yt_32 X_mean_32 y_mean_32 X_scale_32 sqrt_sw_32Xt_64yt_64 X_mean_64 y_mean_64 X_scale_64 sqrt_sw_64Xt_3264yt_3264 X_mean_3264 y_mean_3264 X_scale_3264 sqrt_sw_3264Xt_6432yt_6432 X_mean_6432 y_mean_6432 X_scale_6432 sqrt_sw_6432s' rtest_dtype_preprocess_datar2sT ))   2 3CIJ J'A A ) q B ::arzz *D ::arzz *D JJr ,E ::arzz *D ::arzz *D JJr ,EAQ  #' B>E5)Y JBR  #' B>E5)Y J   '+  KGWk; l   '+  KGWk; l ;;"** $$ $ ;;"** $$ $ ??bjj (( ( ??bjj (( (   rzz )) )2::--- ;;"** $$ $ ;;"** $$ $ ??bjj (( ( ??bjj (( (   rzz )) )2::--- ==BJJ && & ==BJJ && &    ** *    ** *    ++ +!!RZZ/// ==BJJ && & ==BJJ && &    ** *    ** *    ++ +!!RZZ/// :: ## # :: ## # :: ## # :: ## #E5t$7E5t$7Iyt4Iyt4J + JT:r! n_targetsrctjj|}d}d}d|j|z}|j||}||j|}n|j||}tj|} || ddtj fz} ||| z} n|| ddtj fz} |+||}|||j dd}n||}t|||\} } }t|| |2| j} | j} || j} t| | t| | y)Nrkrr(r) r,r-r.rcrnewaxisreshaper r rrd)rr"r5r6r7r8r+rr9expected_sqrt_swexpected_rescaled_Xexpected_rescaled_y rescaled_X rescaled_yrs rtest_rescale_datars^ ))   2 3CIJ#((9--M J'A HHY  HHY *ww}-.q"**}=="22"21bjj="AA# Q    2q!12A #A&3Aq-&H#J GG-.#'') '')  #))+JJ 34J 34r!cXt}|jjtj}|j jtj}||}tj |jtj}|jjtj}|j jtj}||}tj |jtj} t|||\} } t||| \} } | j\} }} } | j\}}} } | \}} } |\}} } |jtjk(sJ|jtjk(sJt||tt|||\}} t||| \}} |j\}}} } |j\}}} } |\}} } |\}} } |jtjk(sJ|jtjk(sJt||tt||tt||t||t||t||y)Nrr)rrastyper,rtargetaranger'rr _next_pyrr rr)riirisrrX_csr_32sample_weight_32rrX_csr_64sample_weight_64 dataset_32r dataset_64xi_32yi_32xi_64yi_64 xi_data_32 xi_data_64 datasetcsr_32 datasetcsr_64xicsr_32yicsr_32xicsr_64yicsr_64 xicsr_data_32 xicsr_data_64s rtest_fused_types_make_datasetr4s9 ;D 99  BJJ 'D ;;  bjj )DT"Hyy"**= 99  BJJ 'D ;;  bjj )DT"Hyy"**=!t-=>MJ t-=>MJ#,,.E5!Q#,,.E5!QJ1J1   rzz )) )   rzz )) )E5t,$Hd4DEM1#Hd4DEM1,557Hh1,557Hh1"M1a"M1a   "** ,, ,   "** ,, ,M=t<HhT2z=1z=1uh'uh'r!X_shape))rCr&)rC)rbrbz/Known to fail for CSR arrays, see issue #30131.)reason)marksctjj|}|\}}|j||}|j|}|||}t |} t di| j ||d} | jj} |r | j} tj|} | j ||| t| j| d|rt| j d} | j ||| t| j| d|rt| j |jdd|jd  } | j ||| } | jj} |r | j} | j ||tj| zt| j| |dnd |rt| j | j}d |d d|d dxxxd zccc| j |||| jj}|r | j}| j |dd |dd | dd t| j|d |rt| j| tj ||d|dzgd}ntj"||d|dzgd }tj"||d|dzg}| j}|d|dzxxxdzccctj"| | d|dzgd }t di| j |||}t di| j |||}t|j|jd|r!t|j|jyy)zTest that the impact of sample_weight is consistent. Note that this test is stricter than the common test check_sample_weight_equivalence alone and also tests sparse X. It is very similar to test_enet_sample_weight_consistency. NrWr*rrg^@g{Gz?rr)lowhighr'gh㈵>icsc)rrrm)r,r-r.rcrnrrrrr ones_liker r0r1pirrz concatenate)r5r"r#r5r6r7r8rr9rqrcoef interceptr+sample_weight_0coef_0 intercept_0rZy2sample_weight_1sample_weight_2reg1reg2s r0test_linear_regression_sample_weight_consistencyrKsd0 ))   2 3C#Iz J'A A# Q   .F  $V $ ( (AT ( BC 99>> DNN LLOMGGAq G.CIIt$/ 2MGGAq G.CIIt$/ 2KKDqqwwqzKBM ''!Qm' 4C 99>> DNN GGAq 5G6CIIt2B2J$PTU 2$((*OOBCbcFdNFGGAqG0 YY^^ Fnn GGAcrFAcrF-*23E B ^^Q"2IN 341 = A. Q/0 1B#((*O$i1n%*%nn  &6 Q78qO  %f % ) )!Qo ) ND  %f % ) )"b ) PDDJJ 69r!):rnumpyr,rJscipyrrsklearn.datasetsrrrsklearn.linear_modelrsklearn.linear_model._baser r r sklearn.preprocessingr sklearn.utils._testingr rrsklearn.utils.fixesrrrrrr mark parametrizerArNrUr`rhrtr}rrrrrrrrrrrrrrrr4paramxfailrK) containers0rrYs) QQ1 4  30+dVn-DE4-8$+9F$+N 0662BC'D'"P& =4-8.97:97$P".9P:P").;" 99"+dVn-DED%=1382F38l087<+dVn-DE.F.**UDM:+dVn-DEF1F;F1R.9:::<.9 : +dVn-DEM2*3F***UDM:5$-8^;9;^;BtQi0+dVn-DE&5F1&5R.9*(:*(Z$CD F(   ++##H$   5$-8Q:9 EQ:s"5O