Pi\ddlZddlZddlZddlmZddlmZmZm Z ddl m Z ej j ZedddfZej\ZZdZdZdZdZd Zd Zd Zd Zejd dgdZdS)N)datasets) MinCovDetempirical_covariancefast_mcd)assert_array_almost_equalc ,tdddddd|tdddddd |tddd ddd |td dd ddd|tdddddd|tdddddd|tdddddd|dS)Ndrg{Gz?皙?J333333?A(2iiiii ifii^)launch_mcd_on_dataset)global_random_seeds /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/sklearn/covariance/tests/test_robust_covariance.pytest_mcdrs#q!T34FGGG#q"c34FGGG#q"c34FGGG$3S#7IJJJ$3S#7IJJJ#q#tT38JKKK"b!S#r3EFFFFFctjd}d}tjt|5t |ddddS#1swxYwYdSNr z'Expected 2D array, got 1D array insteadmatch)nparangepytestraises ValueErrorr)Xmsgs rtest_fast_mcd_on_invalid_inputr$)s #A 3C z - - - sAAActjd}t}d}tjt |5||ddddS#1swxYwYdSr)rrrrr r!fit)r"mcdr#s rtest_mcd_class_on_invalid_inputr(0s #A ++C 3C z - - -  sA##A'*A'c6tj|}|||}||d|} d|d||fdz z} || xx| z cc<tj|t} d| | <|| } t| |} | j }| j }| j }tj| d|z dz}||ksJtjt| |z dz}||ksJtj||ksJt#| || jdS)Ng$@size?F random_stater)rrandom RandomStaterandn permutationrandintonesastypeboolrr& location_ covariance_support_meanrsumr mahalanobisdist_) n_samples n_features n_outlierstol_loctol_cov tol_supportseedrand_gendataoutliers_indexoutliers_offset inliers_mask pure_datamcd_fitTSHerror_location error_covs rrr8sy$$T**H >>)Z 0 0D)))44[j[ANh..q J7O.PPSVVWOO+79%%,,T22L#(L \"IT***..t44GAAAWinnQ//!39::N G # # # #-i881<BCCI w     6!99 # # # #g11$77GGGGGrctjd}|d}t }||dS)Nr)rr+)rr0r1normalrr&)rndr"r's rtest_mcd_issue1127rVSsH )   " "C  A ++CGGAJJJJJrctj|}tjddd}tjt tj||}tj |tj |j ddff}t| |dS)Nr rrr.)rr0r1linspacetolistarraylist itertoolsproducthstackzerosshaperr&)rrF data_valuesrGs rtest_mcd_issue3367rd\sy$$%788H+b!R((//11K 8D*; DDEE F FD 9dBHdjmQ%7889 : :D"8$$$((.....rcptjgd}|dd}tjgd}|dd}d}||fD]V}tjt |5t |dddn #1swxYwYWdS)N) r-r r r 9v?r r r Cl?r r) r-rrrrfrrrrgrzYThe covariance matrix of the support data is equal to 0, try to increase support_fractionr)rr\reshaperr r!rr&)X_1X_2r#r"s r#test_mcd_support_covariance_is_zerorl|s (JJJ K KC ++b!  C (JJJ K KC ++b!  C $3Z ]:S 1 1 1   KKOOA                  s<"B**B. 1B. c4gdgdgdgdgdgdgdgdgd gd gd gd gd gdgdgdgdgdgdg}td|}d}tjt|5||ddddS#1swxYwYdS)N)ffffff@ @ffffff?皙?)皙@@rprq)g@g @g?rq)ffffff@@?rq)@ @rprq)rt333333 @rpr)rwryrvrq)g@g333333@rprq)rrrurvr )皙@g @rvrq)333333@ry皙?rq)r{rsrpr )g333333@rsg?r )rnrorpr)g@gffffff@333333?r)rzryr}rq)rtrx?rq)rwrsr|rq)g@rorvrqr-)support_fractionr/zDeterminant has increasedr)rrwarnsRuntimeWarningr&)rr"r'warn_msgs rtest_mcd_increasing_det_warningrsX ' A, S7I J J JC*H nH 5 5 5  s*B  BBzn_samples,n_features)irYcd}tj|}|||f}t |jd}tddd||jj }tj |}tj |tj |z } | |ks Jdd S) zCheck that MinCovDet does not underestimate the empirical variance on Gaussian data. A large sample size and n_features makes the test robust. Non-regression test for: https://github.com/scikit-learn/scikit-learn/issues/23162 gQ?r+T)assume_centeredr~F)rstore_precisionrr/z.MinCovDet underestimates the Gaussian varianceN) rr0 default_rngrTrrMdiagonalrr&r9diagr<) r?r@r thresholdrngxvar_empcov_mcdvar_mcdmean_var_ratios rtest_mincovdet_bias_on_normalrsI )   2 3 3C Y/ 00A"13===FFHHG ! +    QS  ggGVG__rvg6N I % % %'W % % % % %r)r^numpyrrsklearnrsklearn.covariancerrrsklearn.utils._testingr load_irisrGr"X_1drbr?r@rr$r(rrVrdrlrmark parametrizerrrrsZ HHHHHHHHHH<<<<<<HAw :GGG.HHH6///@    D/*>>XX?>XXXr