{Kg]jddlZddlZddlZddlmZmZddlmZm 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 dgezd%Z0ej<j?d&ed'Z1ej<j?ded(Z2ej<j?d dgezej<j?d)d*d+Z3d,Z4ej<j?d-dd.gej<j?d dgezd/Z5ej<j?ded0Z6ej<j?d dgezej<j?d dd gd1Z7y)2N)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 h/home/alanp/www/video.onchill/myenv/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 |} | j |||| j} | j} | jj|jdfk(sJtj|} |s|n t|} | j| z| z}| j| z|z}tj||}|s t| |yt| |ddt| |dy)N))size?r# sample_weightrr)nprandom RandomStatenormaluniformrrrrshapediagr Trsolver )r"r#global_random_seedrng n_samples n_featuresryr+rcoefs1inter1WX_augXwywcoefs2s r%test_linear_regression_sample_weightsrA<s< ))   2 3C!Iz J/ 0A# Q  "A#++9+55M  7CGGAq G. YYF ^^F 99??qwwqzm ++ +  A"A(9!(?ckk!nqwwy8!O>O>Q6RR'(]## Q '7  # '#BFFG fo6fo6grwwz':;#!"**, <!"a/&9:b!o"56r! lil_containerc<tjj|}d}d}tj||d|}||}|j |}|j }t ||d\}} } } } t| tj|t| dt| tj|t|j |t| |t ||d\}} } } } t| tj|d t| tj|d t| tj|t|j |t| |tj|d z y) Nrjrr)densityrvFr)rTr) r,r-r.rrbrr rrrIr) r5rr6r7r8rr9XArrrrrs r#test_sparse_preprocess_data_offsetsrs; ))   2 3CIJ Iz3SIAaA A B&6q!5&Q#BFFGfbhhz&:;fa(grwwz':;bjjlB/b!$&6q!4&P#BFFGfbggbq&9:fbggaa&89grwwz':;bjjlB/b!bggaa&8"89r!ct\}}d||dk<||}t||d\}}}}}|jdk(sJy)Nrk@Tr)csr)rr format)rhrr9rcsr_rs rtest_csr_preprocess_datarsR  DAqAa#gJ  C'QdCD!Q1 ;;%  r!to_copyTFct\}}d||dk<|||}t||d|d\}}}}}|r/|-tj|j|jrJy|rtj||rJy|-tj|j|jsJytj||sJy)NrkrTF)r#r check_input)rr r,may_share_memorydata)r"rrr9X_y_rs r#test_preprocess_copy_data_no_checksrs  DAqAa#gJ# Q & 1DwEOBAq!#/&&rww7777 &&r1----  %""277AFF333""2q)))r!ctjj|}d}d}|j||}|j|}tj|tj }tj|tj }tj|tj }tj|tj } dD]w} t||| \} } } }}t|| | \}}}}}t|| | \}}}}}t||| \}}}}}| jtj k(sJ| jtj k(sJ| jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ|jtj k(sJ| jtj k(sJt| |t| |t| |t||t||zy)Nrjrrrr)) r,r-r.rbasarrayfloat32float64r rr)r5r6r7r8rr9X_32y_32X_64y_64r#Xt_32yt_32 X_mean_32 y_mean_32 X_scale_32Xt_64yt_64 X_mean_64 y_mean_64 X_scale_64Xt_3264yt_3264 X_mean_3264 y_mean_3264 X_scale_3264Xt_6432yt_6432 X_mean_6432 y_mean_6432 X_scale_6432s rtest_dtype_preprocess_datarss ))   2 3CIJ J'A A ::arzz *D ::arzz *D ::arzz *D ::arzz *D& 9I  ': 6uiJ :J  ': 6uiJ DT  'D @+{L DT  'D @+{L {{bjj((({{bjj((("**,,,"**,,,2::---{{bjj((({{bjj((("**,,,"**,,,2::---}} ***}} ***  BJJ...  BJJ...!!RZZ///}} ***}} ***  BJJ...  BJJ...!!RZZ///zzRZZ'''zzRZZ'''zzRZZ'''zzRZZ'''!%/!%/!)Y7!)Y7!*j9u'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)Nrjrr(r) r,r-r.rbrnewaxisreshaper r rrc)rr"r5r6r7r8r+rr9expected_sqrt_swexpected_rescaled_Xexpected_rescaled_y rescaled_X rescaled_ysqrt_sws rtest_rescale_datar as^ ))   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)Nrrtol)rrastyper,rtargetaranger'rr _next_pyrr rr)rhirisrrX_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_datasetr(s9 ;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!ctjj|}d\}}|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 dxxxdzccc| j ||| | jj}|r | j}| j |dd |dd | dd |r|n0t| j|d |rt| j| tj ||d|d zgd}ntj"||d|d zgd }tj"||d|d zg}| j}|d|d zxxxd zccctj"| | d|d zgd }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_weights_invariance alone and also tests sparse X. It is very similar to test_enet_sample_weight_consistency. )rCr&Nr)r*rr g^@g{Gz?rr)lowhighr'gh㈵>icsc)rrrl)r,r-r.rbrmrrrrr ones_liker r0r1pirry concatenate)r"r#r5r6r7r8rr9rprcoef interceptr+sample_weight_0coef_0 intercept_0rYy2sample_weight_1sample_weight_2reg1reg2s r0test_linear_regression_sample_weight_consistencyr;sr ))   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!)8rnumpyr,rJscipyrrsklearn.datasetsrrrsklearn.linear_modelrsklearn.linear_model._baser r r sklearn.preprocessingr sklearn.utils._testingr rrsklearn.utils.fixesrrrrrr mark parametrizerArNrUr_rgrsr|rrrrrrrrrrrrrrr r(r;rlr!rrFs` QQ1 4  30+dVn-DE4-8$+9F$+N 0662BC'D'"P& =4-8.97:97$P".9P:P").;" 99"+dVn-DED%=1382F38l0870+dVn-DE.F.*+dVn-DE67F67r.9:::0.9 : +dVn-DEM2*3F**F:RtQi0+dVn-DE&5F1&5R.9*(:*(Z+dVn-DE5$-8Y:9FY:r!