h$,f<dZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZddlmZddlmZd!dZd Zd Zd Zej.j1d d dgdZdZdZej.j1d d dgdZdZej.j=dej.j1dgddZdZ dZ!ej.j1dgdej.j1dd dgej.j1dd dgdZ"ej.j1d d dgdZ#d Z$y)"z Test the fastica algorithm. N)stats)PCAFastICAfastica)_gs_decorrelation)ConvergenceWarning)assert_allclosectj||}||jdz}||jdz}y)aCenters and norms x **in place** Parameters ----------- x: ndarray Array with an axis of observations (statistical units) measured on random variables. axis: int, optional Axis along which the mean and variance are calculated. raxisN)nprollaxismeanstd)xr s Hlib/python3.12/site-packages/sklearn/decomposition/tests/test_fastica.pycenter_and_normrs< AtAQAAActjjd}tjj |j dd\}}}|j d}t ||d|dzjdksJ|j d}t ||d}tj||j}|dddzjdksJy)Nr g|=) r random RandomStatelinalgsvdrandnrsumdotT)rngW_wutmps rtest_gsr'"s ))   "CiimmCIIb"-.GAq! " AaB qD::<' !! ! " A!Q"A &&ACC.C GqL   ' )) )rctjjd}|jdj |d}t ddddj |}|jj|k(sJ|jj|k(sJ|jj|k(sJ|jj|k(sJy) NrdrFcopyr unit-variance n_componentsmax_iterwhiten random_state) r rr random_sampleastyperfit components_dtypemixing_mean_ whitening_) global_dtyper!Xficas rtest_fastica_attributes_dtypesr?0s ))   "C )$++Lu+EA oA  c!f     ! !\ 11 1 <<   -- - ::  | ++ + ?? L 00 0rctjjd}|jdj |d}t |dd|\}}}|j |k(sJ|j |k(sJ|j |k(sJy)Nrr)Fr+r-r.)r1r2r3)r rrr4r5rr8)r<r!r=k_r9s_s rtest_fastica_return_dtypesrC<s ))   "C )$++Lu+EA DsOB 88| ## # ==L (( ( 88| ## #r add_noiseTFc  |dk(rB|tjk(r/|s-tjddk(rt j dtj j|}d}dtjtjdd|zdkDd z }tjjd || }tj||fj}t||j!|}|\}}d }tj"tj$|tj|gtj|tj$| gg} | j!|} tj&| |} |r| d |j)ddzz } t| d } ddg} ddd| g} gd}t+j,| | |D]\}}}|rht/| j||||\}}}t j0t25t/| jtj4||dddn{t7dd|}|j9| j}t/|||d|\}}}t j0t25t/|tj4|ddd|j}|rM|tjk(rdnd}t;tj&tj&||| ||t||\}}t=tj&||t=tj&||kDr|\}}|tj>tj&||z}|tj>tj&||z}|sLt;tj&|||z d dt;tj&|||z d dBt;tj&|||z d d t;tj&|||z d d t/| j|\}}}tA|||}|j9| j}|jBjDdk(sJ|jDd k(sJt;||tjFtj<||tjk(rdnd!z}t;||jI| j||jJjDdk(sJtAtj4|}t j0t25|jM| jdddy#1swYxYw#1swYxYw#1swYyxYw)"NDISTRIBubuntuzFastICA instability with Ubuntu Atlas build with float32 global_dtype. For more details, see https://github.com/scikit-learn/scikit-learn/issues/24131#issuecomment-1208091119r-rrr*sizer3g333333?皙?c<|dzd|dzzjdfS)Nrr )r)rs rg_testz#test_fastica_simple..g_testls$!ta!Q$h__"_---rparallel deflationlogcoshexpcubearbitrary-variancer.F)funr2 algorithmr3)rXr2rYTr0r2r3F)rXrYr2r3)rXrYgh㈵>atolg{Gz?)rXrYr3rrr-rgHz>)'r float32osgetenvpytestxfailrrsinlinspacertrvsc_r rr5arraycosrr itertoolsproductrraises ValueErrortanhr fit_transformr abssignrr7shapemax transformr9r6)rDglobal_random_seedr<r! n_sampless1s2sphimixingmrPalgosnls whiteningalgonlr2rAr9rBpcar=r\s1_s2_r# sources_funicasourcess rtest_fastica_simplerGs b BJJ & IIi H ,  `  ))   2 3CI bffR[[C34 4q 8A =B QY5G HB b"f AA A FB C XXs RVVC[1BFF3K"&&+3NO PF ]]< (F vqA S399Q% %%A. %E eVV ,C>I%--eS)D)Gb& %FdOBz* I4H I I1TDC!!!##&A%rT%cOBz* 8rww$7 8 TT  (2::541D BFF266'2#6:BT JS rvvc2 #bffS"o"6 6HC rwwrvvc2'' rwwrvvc2'' BFF3Oi7 F BFF3Oi7 F BFF3Oi7 F BFF3Oi7 FS)GX  t2DAq+ bD7I JC$G ?? F ** * ==I %% %K) 66"&&/ "lbjj.Hdd SDGS]]133/d; ;;   && & bgg .C z "  q I I 8 8`s$(W$WWW W W'cddgddgg}tddd}d}tjt|5|j |dddt |dsJy#1swYxYw)NrrIFrZz(Ignoring n_components with whiten=False.matchr9)rrbwarns UserWarningr6hasattr)r}rwarn_msgs rtest_fastica_nowhitenrsg Q!QA qQ ?C9H k 2   3 "" "s AA&cntjjd}d}tjdd|}tj|}tj tjtj |z}tj||fj}t||jdd}tj||}d}tjt|5tdd|dd } | j!|jdddy#1swYyxYw) Nrr-r*rz\FastICA did not converge. Consider increasing tolerance or the maximum number of iterations.rrQ)rYr0r3r1tol)r rrrerdceilpirhr rrrrbrrrr6) r!rwrfrxryrzr|r}rrs rtest_fastica_convergence_failrs ))   "CI AsI&A B  " #B b"f AAYYq!_F vqA /  ( 9 qsQTW   s 6,D++D4c tjjd}d}tjdd|}tj|}tj tjtj |z}tj||fj}t||\}}|jdd}tj||}|r|d|jd|zz }t|t|jdd|\} } } | j} t| tjtj| | |t| | \} } ttj| |ttj| |kDr| \} } | tjtj| |z} | tjtj| |z} |sKttj| ||z d d ttj| ||z d d yy) Nrr-r*rrrLr.rZrIgMbP?r[)r rrrerdrrrhr rrrrr rqrr)rDr!rwrfrxryrzr|r}rAr9rBrrs rtest_non_square_fasticars ))   "CI AsI&A B  " #B b"f AA FBYYq!_F vqA S399Q * **A !O#OB BBrvvgr2A67BHC 266#r?c"&&b/22S277266#r? ##C277266#r? ##C sB)3QTBsB)3QTB rctjj|}|jdj |}d}ddgddgfD]\}}||n|j d}t |||d }tj5tjd ttjd t|j|} ddd|jj |d fk(sJ j |j d|fk(sJt |||d } tj5tjd ttjd t| j|ddd| jj |d fk(sJ| j|} |r'tj | j#d z } nd} t%| | | y#1swY#xYw#1swYxYw)zTest unit variance of transformed data using FastICA algorithm. Check that `fit_transform` gives the same result as applying `fit` and then `transform`. Bug #13056 r)i,r.rFNrIrr/errorignorerg.Arr[)r rrr4r5rsrwarningscatch_warnings simplefilterRuntimeWarningrrpr7r6rurqrr ) rvr<r!r=r1r2r0 n_components_rXtica2Xt2r\s rtest_fit_transformr s ))   2 3C )$++L9AH"11!5t} E#, (4(@ aggaj %VW  $ $ & &  ! !'> :  ! !(,> ?""1%B &$$(;;;;xxAGGAJ 6666%VW  $ $ &   ! !'> :  ! !(,> ? HHQK   %%-)<<<<nnQ 66#;##%+DDCd+G#,  & &  sAG.5AG;.G8 ;H z/ignore:Ignoring n_components with whiten=False.z+whiten, n_components, expected_mixing_shape))rWrrr)rWrrr)r.rr)r.rr)Frr)Frrctd}tjj|}|j|dfj |}t |||}t j5t jdt|j|} ddd|jj|k(sJ|j } |j| jk(sJ||jdk(r:|r'tj| jdz } nd} t!|| | yy#1swYxYw) Nr*r)r0r3r2rrIgj@rr[)r rrr4r5rrrrrrpr9rsinverse_transformrqrr ) r2r0expected_mixing_shapervr<rwr!r=rrX2r\s rtest_inverse_transformr<s I ))   2 3C 9b/*11,?A |#f MC  "" h(:;   q ! " ;;   5 55 5  r "B 77bhh  qwwqz! 66":??$s*DD2D)"""s &,D..D7cd}d}tjjd}|j||f}|j |dz|dz}t j td5t|ddidddt j td 5t|| dddy#1swY;xYw#1swYyxYw) NrNrrrIzalpha must be in \[1,2\]ralpha)fun_argsz0w_init has invalid shape.+should be \(3L?, 3L?\))w_init) r rrr4rrbrmrnr) n_featuresrwr!r=rs rtest_fastica_errorsrhsJI ))   "C 9j12A YYzA~zA~ 6F z)D E*WaL)* P " &!""**""s*B4C4B=C ctjjd}|jd}|jd}t |dd}|j |}tj|tjdk(sJy)zTTest unit variance of transformed data using FastICA algorithm. Bug #13056 rr)rIr.rZg?N) r rrr4rsrrpvarrbapprox)r!r=r0rrs r!test_fastica_whiten_unit_variancervss ))   "C )$A771:L |ORS TC  1 B 66":s+ ++ +rr2rV return_X_mean return_n_itercd}d}tjjd}|j||f}d|z|z}t ||||}t ||k(sJ|s|dJyy)NrNrr)r2rr)r rrr4rlen) r2rrrrwr!r= expected_lenouts rtest_fastica_output_shapersJI ))   "C 9j12A}$}4L  & ] C s8| ## # 1v~~ rctjj|}d}dtjtjdd|zdkDdz }t j jd||}tj||fj}t||\}}|jdztjz}tjtj|tj|gtj|tj| gg}tj||} |r| d|j!ddzz } t| i} dD]\} t#dd | } | j%| j} | | | <| j&j(d k(sJ| j(d k(r\Jt+| d | ddy)z2Test FastICA is consistent between whiten_solvers.r-rrr*rIrJrL)reighr.r3r2 whiten_solverr]r^rrg-q=r[N)r rrrdrerrfrgrhr rrandrrirjrrrrpr7rsr )rDrvr!rwrxryrzr{r|r}outssolverrrs r%test_fastica_simple_different_solversrs ))   2 3CI bffR[[C34 4q 8A =B QYS 9B b"f AA FB ((*q.255 C XXs RVVC[1BFF3K"&&+3NO PF vqA S399Q% %%A D!*1_FS##ACC(V $$...}} ))) *DL$u+E:rc&tjj|}|jdd}||jz}t ddd}d}t jt|5|j|d d d y #1swYy xYw) z:Test FastICA eigh solver raises warning for low-rank data.rrrr.rrz$There are some small singular valuesrN) r rrrr rrbrrr6)rvr!Ar=rmsgs r"test_fastica_eigh_low_rank_warningrsu ))   2 3C "aA ACCA q OC 0C k -  s ,BB)rO)%__doc__rkr`rnumpyr rbscipyrsklearn.decompositionrrrsklearn.decomposition._fasticarsklearn.exceptionsrsklearn.utils._testingr rr'r?rCmark parametrizerrrrrfilterwarningsrrrrrrrrrs  77<12  * 1$tUm4h5hV#<tUm4*C5*CZ/,dMN1 * O*@ " ,#QR4-84-899S"tUm4;5;Dr