h$,f#ddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZdZdZd Zd Zd Zd Zd Zej,j/dedZej,j/dddgdZdZy)N)mutual_info_classifmutual_info_regression) _compute_mi)check_random_state)assert_allcloseassert_array_equal)CSR_CONTAINERSctjgd}tjgd}dtjdzdtjdzz x}}dtjdzdtjdzz dtjdzz }||z|z }tt ||dd |y) N)rr rr)r rrrr g333333g333333?g?gɿg?T x_discrete y_discrete)nparraylogrr)xyH_xH_yH_xyI_xys Plib/python3.12/site-packages/sklearn/feature_selection/tests/test_mutual_info.pytest_compute_mi_ddrs !A !A266%=(ERVVE]+BBBC# BFF5M !EBFF5M$9 9EBFF5Ms" I Q C = KKYK '! + HHY -Av++b!"&&,+7$;;q!"&&$-;84% UbffSAE]+ +a"&&q/.A ABFF3K O FF1I % =K$1%[J Jt <  ==rcd}tjj|dkD}tj||}|dk(}tjjddtj|||<tjjddtj|||<t ||dd }tj |df}tj |d f}t ||dd }t||y) Ndr rrr>r rTFr r)rrandomr?r@rArhstackr)r2rBrrrDmi_1mi_2s rtest_compute_mi_cd_unique_labelrLesI y)C/A L)A 6DiiABFF4L9AdGyy  ABFFD5M :AteH q! ?D 1a&A 1b'A q! ?DD$rctjgdgdgdgdgdg|}tjgd}t||d}ttj| tjgdy) Nrrrr r rrrr dtyperr rrr Tdiscrete_features)rrr )rrrrargsort)r2Xrmis r!test_mutual_info_classif_discreterYys]  Iy)Y?| A !A QT :Brzz2#(; G G" %G/ 2G; >H correlatedTFchtjj|}d}|jd|}|r |j tj }n|j dd|}t|dddf|dg|}t|dddf|d g|}|tj|k(sJy) zCheck that `mutual_info_classif` and `mutual_info_regression` are symmetric by switching the target `y` as `feature` in `X` and vice versa. Non-regression test for: https://github.com/scikit-learn/scikit-learn/issues/23720 rGrr rr NFrlT) rrH RandomStaterandintr1r`normalrrrmapprox)ryglobal_random_seedr9ndc mi_classif mi_regressions r3test_mutual_information_symmetry_classif_regressionrs ))   2 3C A BQ A HHRZZ  JJq!!J $$ !T' A%?QJ+ !T' A$>PM }5 55 5rc tjj|}|jdd}|j tj d}|jdd}t |||}t |||}t||y)zqCheck that results agree when X is integer dtype and float dtype. Non-regression test for Issue #26696. rG)rGrr Tr"r\N)rrHr{r|r1r`rr)rr9rWX_floatrexpectedresults r'test_mutual_info_regression_X_int_dtypers{ ))   2 3C Ci (Ahhrzzh-G Cc "A%gq?QRH #Aq7I JFFH%r)numpyrrmsklearn.feature_selectionrr&sklearn.feature_selection._mutual_infor sklearn.utilsrsklearn.utils._testingrrsklearn.utils.fixesr rr<rErLrYrargmark parametrizerxrrrrrs Q>,/ O 9F$=N ( =",!0.9+:+Be}56668 &r