U mda@s<ddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZddlmZdZddZejdejejgejdddgddZddZ ejdddgdd Z!d!d"Z"d#d$Z#ejd%ddd&gd'd(Z$d)d*Z%ejdddgd+d,Z&d-d.Z'd/d0Z(d1d2Z)d3d4Z*d5d6Z+d7d8Z,d9d:Z-d;d<Z.d=d>Z/ejd?ddgd@dAZ0dBdCZ1dDdEZ2ejd?dFejdGdFdHdIZ3dJdKZ4ejdLddMgdNdOZ5dPdQZ6dS)RN)sparse)linalg)assert_array_almost_equal)assert_array_equal)assert_allclose)LinearRegression)_deprecate_normalize)_preprocess_data) _rescale_data) make_dataset)make_sparse_uncorrelated)make_regression) load_iris)StandardScaler)add_dummy_featureư>cCsdgdgg}ddg}t}|||t|jdgt|jdgt||ddgdgg}dg}t}|||t|jdgt|jdgt||dgdS)Nr)rfitrcoef_ intercept_predict)XYregr]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/sklearn/linear_model/tests/test_base.pytest_linear_regression s   r array_constr fit_interceptTFcCstj|}d\}}||j||fd}|j|d}d|j|d}t|d} | j|||d| j} | j} | jj |j dfkst t |} |s|nt |} | j | | }| j | |}t||}|st| |n t| |ddt| |ddS)N))size?r sample_weightrr)nprandom RandomStatenormaluniformrrrrshapeAssertionErrorZdiagrTrZsolver)rrglobal_random_seedrng n_samples n_featuresryr&rZcoefs1Zinter1WZX_augZXwZywZcoefs2rrr%test_linear_regression_sample_weights8s&      r5c CsLd}td}td}tdd}tjt|d|||W5QRXdS)Nz7A sparse matrix was passed, but dense data is required. TZpositivematch) reyer'onesrpytestraises TypeErrorr) error_msgrr3rrrr.test_raises_value_error_if_positive_and_sparse_s    r@zn_samples, n_features)r)rArc Csptjd}|||}||}||dd}d}d}t}||||||||||||dS)Nrrrr#g@)r'r(r)randnrr) r1r2r0rr3Zsample_weights_OKZsample_weights_OK_1Zsample_weights_OK_2rrrr9test_raises_value_error_if_sample_weights_greater_than_1dks   rCcCstddgddgg}tdddgdd d gg}td d g}td d ||}t||}td d ||}t||}|jj|jjkst|jj|jjkst|jj|jjkstdS)NgmGB?g\^?gL+K?gfiU?g&/)¶?gsz,test_deprecate_normalize..) FutureWarningappendwarningscatch_warnings simplefilterrr<warnsallr-)rIoutputexpectedZ warning_msg _normalizerrQrtest_deprecate_normalizes$     r^cCsrtj|}d}t||}||}||}t}|||t ||j |j t | ||ddS)Ndr) r'r(r)rr:randrrravelrrrr)r/r0nrbetar3olsrrrtest_linear_regression_sparses   rec Cstjd}d}d}|||}d||dk<t|}||}t|d}tf|}tf|} | ||| |||j t | j kst t|j| jdS)Nrrg?r$)r'r(r)rBr csr_matrixr`dictrrrr<Zapproxr-rr) rr0r1r2rZXcsrr3paramsZ clf_denseZ clf_sparserrr)test_linear_regression_sparse_equal_denses          rkcCstjd}t|d\}}t||fj}|jd}t}||||j jd|fksZt | |}|||| |}t t||fj|dddS)Nr random_staterrrAdecimal) r'r(r)r vstackr.r,rrrr-rr)r0rr3rr2rY_predy_predrrr'test_linear_regression_multiple_outcomes      rsc Cstj|}t|d\}}t|}t||fj}|jd}t }| |||j jd|fksdt | |}| ||| |}tt||fj|dddS)NrlrrrArn)r'r(r)r rZ coo_matrixrpr.r,rrrr-rrar r/r0rr3rr2rdrqrrrrr.test_linear_regression_sparse_multiple_outcomes      rucCsdgdgg}ddg}tdd}|||t|jdgt|jdgt||ddgdgg}dg}tdd}|||t|jdgt|jdgt||dgdS)NrrTr7r)rrrrrrr)rr3rrrrtest_linear_regression_positives     rvc Cstj|}t|d\}}t||fj}|jd}tdd}||||j jd|fks^t t |j dksrt | |}||| | |}tt||fj|dS)NrlrTr7rrg)r'r(r)r rpr.r,rrrr-rZrrarrtrrr0test_linear_regression_positive_multiple_outcomes      rwcCshtj|}t|d\}}tdd}|||tdd}|||t|j|jddksdtdS)NrlTr7FrgMbP?) r'r(r)r rrmeanrr-)r/r0rr3rregnrrr.test_linear_regression_positive_vs_nonpositive s     rzcCstj|}d}d}|||}|dddfd|dddfd|dddfd|dddf}tdd }|||td d }|||t|j|jdd kstdS) NrfrrrrAg?Tr7Fr) r'r(r)r`rrrxrr-)r/r0r1r2rr3rryrrrtdt | |jdddd f|jddd fW5QRXdS)NZpandas0r6rr{r)Z fill_valuez+pandas.DataFrame with sparse columns found.r8rrArrL)r<Z importorskipZ DataFramer'r(rBrangeZarraysZ SparseArrayrMrrY UserWarningrZilochasattrr-rVrWrX)pdZdfcolZarrmsgrrrr2test_linear_regression_pd_sparse_dataframe_warning>s"   6  rcCsrtj|}d}d}|||}||}tj|dd}tj|ddt|jd}tj|dd}t||ddd\} } } } } t | t |t | dt | t |t | |t | |t||ddd\} } } } } t | |t | |t | t |t | ||t | ||t||ddd\} } } } } t | |t | |t | |t | |||t | ||dS)NrfrraxisFrrIT) r'r(r)r`rxstdsqrtr,r rzerosr;)r/r0r1r2rr3expected_X_meanexpected_X_scaleexpected_y_meanXtytX_meany_meanX_scalerrrtest_preprocess_dataZsR           rc Cstj|}d}d}d}|||}|||}tj|dd}|t|g}|D]}t||ddd\} } } } } t| t |t| |t||ddd\} } } } } t| |t| || t||ddd\} } } } } t| |t| || qPdS) NrfrArrrFrT) r'r(r)r`rxrZ csc_matrixr rr) r/r0r1r2Z n_outputsrr3rargs_rrrrr test_preprocess_data_multioutputs8      r is_sparsecCstj|}d}d}|||}d||dk<|dddfd9<d|dddf<d|ddd f<||}||}tj|d|d }tj|d|d } tj||dd } tj|| d|dd } | dt|jjk} t| ddd d gt | t | } d | | <|rt |}t ||d d|d\}}}}}t||t|| t|t||rrt||nt|||t||| t ||d d |d\}}}}}t||t|| t|| |rt||| nt|||| |rtddj||d}t||ddddft | |ddddft||ddddf|ddddfn>td dj||d}t|j|t||t | |t||| dS)Nrfr{rg?rr6r#rrA)rweights)rrrTF)rrIr&)Z with_meanr%)r'r(r)r`ZaverageZfinfodtypeZepsrrsumrrhr rr;toarrayrrZ transformZmean_)rr/r0r1r2rr3r&rrZX_sample_weight_avgZX_sample_weight_varZ constant_maskrrrrrrZscalerrrrtest_preprocess_data_weighteds            * rc Cstj|}d}d}tj||d|d}|}||}|}tj|ddt|j d}t ||ddd\}} } } } t | t |t | dt | t |t |j|t | |t ||d dd\}} } } } t | tj|ddt | tj|ddt | t |t |j|t | |tj|ddt ||d d d\}} } } } t | tj|ddt | tj|ddt | |t |j||t | |tj|dddS) Nrfrr)ZdensityrmrrFrT)r'r(r)rr`Ztolilrrrr,r rrr;Arx) r/r0r1r2rr3ZXArrrrrrrrr#test_sparse_preprocess_data_offsetssR       rcCsJt\}}d||dk<t|}t||d\}}}}}|dksFtdS)Nrg@Tcsr)r rrhr Z getformatr-)rr3rZcsr_rrrrtest_csr_preprocess_data+s    rTFto_copycCst\}}d||dk<|r$t|}t||d|dd\}}}}}|r^|r^t|j|jrtn@|rtt||rtn*|rt|j|jstnt||stdS)NrgrTF)copyZ check_input)r rrhr r'Zmay_share_memorydatar-)rrrr3ZX_Zy_rrrr#test_preprocess_copy_data_no_checks4s   rc Cstj|}d}d}|||}||}tj|tjd}tj|tjd}tj|tjd}tj|tjd} dD]X} dD]L} t||| | d\} } }}}t|| | | d\}}}}}t|| | | d\}}}}}t||| | d\}}}}}| jtjkst | jtjkst |jtjkst |jtjks,t |jtjks>t |jtjksPt |jtjksbt |jtjkstt |jtjkst |jtjkst |jtjkst |jtjkst |jtjkst |jtjkst |jtjkst |jtjkst |jtjkst |jtjks(t |jtjks:t |jtjksLt |jtjks^t |jtjkspt |jtjkst | jtjkst t | |t | |t ||t ||t ||qxqndS)Nrfrrrr) r'r(r)r`Zasarrayfloat32float64r rr-r) r/r0r1r2rr3X_32y_32X_64y_64rrIZXt_32Zyt_32Z X_mean_32Z y_mean_32Z X_scale_32ZXt_64Zyt_64Z X_mean_64Z y_mean_64Z X_scale_64ZXt_3264Zyt_3264Z X_mean_3264Z y_mean_3264Z X_scale_3264ZXt_6432Zyt_6432Z X_mean_6432Z y_mean_6432Z X_scale_6432rrrtest_dtype_preprocess_dataIs         r n_targetsrc Cstj|}d}d}d||}|||}|dkrB||}n |||}t|||\}} } || ddtjf} |dkr|| } n|| ddtjf} t|| t| | dS)Nrfrr#)r'r(r)r`r Znewaxisr) rr/r0r1r2r&rr3Z rescaled_XZ rescaled_yZsqrt_swZ rescaled_X2Z rescaled_y2rrrtest_rescale_data_denses      rcCst}|jtj}|jtj}t|}tj|j tjd}|jtj }|jtj }t|}tj|j tj d}t |||\} } t |||\} } | \} } } } | \}}} } | \}} } |\}} } |j tjkst|j tj kstt|| tdt |||\}} t |||\}} | \}}} } | \}}} } |\}} } |\}} } |j tjks^t|j tj ksptt||tdt||tdt||t||t| |t||dS)Nr)rtol)rrZastyper'rtargetrrhZaranger"rr Z_next_pyrr-rrr)ZirisrrZX_csr_32Zsample_weight_32rrZX_csr_64Zsample_weight_64Z dataset_32rZ dataset_64Zxi_32Zyi_32Zxi_64Zyi_64Z xi_data_32Z xi_data_64Z datasetcsr_32Z datasetcsr_64Zxicsr_32Zyicsr_32Zxicsr_64Zyicsr_64Z xicsr_data_32Z xicsr_data_64rrrtest_fused_types_make_datasets@         r)7r<rVnumpyr'ZscipyrrZsklearn.utils._testingrrrZsklearn.linear_modelrZsklearn.linear_model._baserr r r Zsklearn.datasetsr r rZsklearn.preprocessingrrrrmarkZ parametrizerDrhr5r@rCrFrJr^rerkrsrurvrwrzr|rrrrrrrrrrrrrrsh                %     & h&   O