U ‘Ç-e<ã @s‚dZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZddlmZddlmZd7d d „Zd d „Zd d„Zdd„Zej dddg¡dd„ƒZdd„Zdd„Zej dddg¡dd„ƒZdd„Zej d¡ej dd d!d"d#d$d%g¡d&d'„ƒƒZd(d)„Z d*d+„Z!ej d,d-d.dg¡ej d/ddg¡ej d0ddg¡d1d2„ƒƒƒZ"ej dddg¡d3d4„ƒZ#d5d6„Z$dS)8z Test the fastica algorithm. éN)Ústats)ÚPCAÚFastICAÚfastica)Ú_gs_decorrelation)ÚConvergenceWarning)Úassert_allcloseéÿÿÿÿcCs0t ||¡}||jdd8}||jdd}dS)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. r©ÚaxisN)ÚnpZrollaxisÚmeanZstd)Úxr ©rúi/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/sklearn/decomposition/tests/test_fastica.pyÚcenter_and_norms rcCs”tj d¡}tj | dd¡¡\}}}| d¡}t||dƒ|d ¡dksPt‚| d¡}t||dƒ}t  ||j ¡}|dd…d ¡dkst‚dS)Nré ég»½×Ùß|Û=é) r ÚrandomÚ RandomStateZlinalgÚsvdÚrandnrÚsumÚAssertionErrorÚdotÚT)ÚrngÚWÚ_ÚwÚuÚtmprrrÚtest_gs"s     r#cCsztj d¡}| d¡j|dd}tddddd |¡}|jj|ksFt ‚|j j|ksVt ‚|j j|ksft ‚|j j|ksvt ‚dS) Nr©édrF©Úcopyréèú unit-variance©Ú n_componentsÚmax_iterÚwhitenÚ random_state) r rrÚ random_sampleÚastyperÚfitÚ components_ÚdtyperÚmixing_Zmean_Z whitening_)Ú global_dtyperÚXZficarrrÚtest_fastica_attributes_dtypes0s ÿþr7cCsdtj d¡}| d¡j|dd}t|dd|d\}}}|j|ksDt‚|j|ksRt‚|j|ks`t‚dS)Nrr$Fr&r(r))r,r-r.)r rrr/r0rr3r)r5rr6Úk_r4Ús_rrrÚtest_fastica_return_dtypes<s ÿ r:Ú add_noiseTFc CsÞ|dkr.|tjkr.|s.t d¡dkr.t d¡tj |¡}d}dt t  dd|¡¡dkd }t j j d ||d }tj ||fj}t|ƒ| |¡}|\}}d }t t |¡t |¡gt |¡t |¡ gg¡} |  |¡} t | |¡} |rü| d | dd¡7} t| ƒd d„} ddg} ddd| g} dddg}t | | |¡D]Ä\}}}|rt| j||||d\}}}t t¡t| jtj||dW5QRXnXtdd|d}| | j¡}t|||d|d\}}}t t¡t|tj|dW5QRX|j}|r&|tjkrdnd}tt t ||¡| ¡||dt|ƒ|\}}tt ||¡ƒtt ||¡ƒkr`|\}}|t t ||¡¡9}|t t ||¡¡9}|sÈtt ||¡|d ddtt ||¡|d ddn4tt ||¡|d d dtt ||¡|d d dq8t| j|||d \}}}t |||d }| | j¡}|j!j"d!ksDt#‚|j"d"ksTt#‚t||ƒt $t |¡¡|tjkr|dnd#}t|| %| j¡|d|j&j"d!ksªt#‚t tj|d}t t¡| '| j¡W5QRXdS)$NéZDISTRIBZubuntuz®FastICA 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%é©Úsizer.g333333ã?çš™™™™™¹?cSs|dd|djddfS)Nérr r )r )rrrrÚg_testlsz#test_fastica_simple..g_testÚparallelZ deflationZlogcoshÚexpZcubeúarbitrary-variancer)F)Úfunr-Ú algorithmr.)rFr-rGT©r+r-r.)rFrGr-r.)rFrGgñh㈵øä>©Úatolg{®Gáz„?)rFrGr.©rr©r(rgH¯¼šò×z>)(r Zfloat32ÚosÚgetenvÚpytestZxfailrrÚsinÚlinspacerÚtÚrvsÚc_rrr0ÚarrayÚcosrrÚ itertoolsÚproductrÚraisesÚ ValueErrorÚtanhrÚ fit_transformrÚabsÚsignrr2ÚshaperÚmaxÚ transformr4r1)r;Úglobal_random_seedr5rÚ n_samplesÚs1Ús2ÚsÚphiÚmixingÚmrBZalgosZnlsZ whiteningÚalgoÚnlr-r8r4r9Zpcar6rJÚs1_Ús2_rZ sources_funÚicaÚsourcesrrrÚtest_fastica_simpleGs¨ÿþý üÿ   0    ÿ    ÿ  "ÿ   $ rpc CsXddgddgg}tdddd}d}tjt|d| |¡W5QRXt|dƒsTt‚dS)Nrr=FrHz(Ignoring n_components with whiten=False.©Úmatchr4)rrOÚwarnsÚ UserWarningr1Úhasattrr)rirnÚwarn_msgrrrÚtest_fastica_nowhiten³s rwc Cs®tj d¡}d}t dd|¡}t |¡}t t tj|¡¡}tj||fj}t |ƒ|  dd¡}t  ||¡}d}t j t|d$tdd|dd d } |  |j¡W5QRXdS) Nrr(r%érz\FastICA did not converge. Consider increasing tolerance or the maximum number of iterations.rqrCç)rGr+r.r,Ztol)r rrrQrPÚceilÚpirTrrrrrOrsrrr1) rrcrRrdrerfrhrirvrnrrrÚtest_fastica_convergence_fail¾s(    ÿÿr|cCsrtj d¡}d}t dd|¡}t |¡}t t tj|¡¡}tj||fj}t |ƒ|\}}|  dd¡}t  ||¡}|rŽ|d|  d|¡7}t |ƒt |jdd|d\} } } | j} t | t  t  | | ¡|¡ƒt | ƒ| \} } tt  | |¡ƒtt  | |¡ƒkr| \} } | t t  | |¡¡9} | t t  | |¡¡9} |snt t  | |¡|d d d t t  | |¡|d d d dS) Nrr(r%rxrr@r)rHr=gü©ñÒMbP?rI)r rrrQrPrzr{rTrrrrrrr]r^)r;rrcrRrdrerfrhrir8r4r9rlrmrrrÚtest_non_square_fasticaÜs<    ÿ "r}c CsZtj |¡}| d¡ |¡}d}ddgddgfD]"\}}|dk rF|n|jd}t|||dd }t ¡(t  d t ¡t  d t ¡|  |¡} W5QRX|j j|d fksªt‚| j|jd|fksÂt‚t|||dd } t ¡(t  d t ¡t  d t ¡|  |¡W5QRX| j j|d fkst‚|  |¡} |rBt | ¡ ¡d } nd} t| | | dq0dS)z¶Test 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)rFNr=rr*ÚerrorÚignorerg€„.AryrI)r rrr/r0r_rÚwarningsÚcatch_warningsÚ simplefilterÚRuntimeWarningrr\r2rr1rar]r r) rbr5rr6r,r-r+Z n_components_rnÚXtZica2ZXt2rJrrrÚtest_fit_transform sB ÿ   ÿ    r…z/ignore:Ignoring n_components with whiten=False.z+whiten, n_components, expected_mixing_shape)rEr©rr)rEr©rr)r)rr†)r)rr‡)Frr‡)Frr‡c CsÂd}tj |¡}| |df¡ |¡}t|||d}t ¡t dt ¡|  |¡} W5QRX|j j |kslt ‚| | ¡} |j | j ks†t ‚||j dkr¾|r¬t | ¡ ¡d} nd} t|| | ddS) Nr%r)r+r.r-rr=gjø@ryrI)r rrr/r0rr€rr‚rr\r4r_rZinverse_transformr]r r) r-r+Zexpected_mixing_shaperbr5rcrr6rnr„ZX2rJrrrÚtest_inverse_transform<s    rˆc CsŠd}d}tj d¡}| ||f¡}| |d|d¡}tjtddt|ddidW5QRXtjtd dt||d W5QRXdS) NrArrr=zalpha must be in \[1,2\]rqÚalpha)Zfun_argsz0w_init has invalid shape.+should be \(3L?, 3L?\))Úw_init) r rrr/rrOrYrZr)Ú n_featuresrcrr6rŠrrrÚtest_fastica_errorshs ÿrŒcCsTtj d¡}| d¡}|jd}t|ddd}| |¡}t |¡t  d¡ksPt ‚dS)zTTest unit variance of transformed data using FastICA algorithm. Bug #13056 rr$r=r)rHgð?N) r rrr/r_rr\ÚvarrOZapproxr)rr6r+rnr„rrrÚ!test_fastica_whiten_unit_variancevs     rŽr-rEr)Ú return_X_meanÚ return_n_iterc Csfd}d}tj d¡}| ||f¡}d||}t||||d}t|ƒ|ksNt‚|sb|ddksbt‚dS)NrArr)r-rr)r rrr/rÚlenr) r-rrr‹rcrr6Ú expected_lenÚoutrrrÚtest_fastica_output_shape„s  ÿr”cCs:tj |¡}d}dt t dd|¡¡dkd}tjjd||d}tj||fj }t |ƒ|\}}|  ¡dtj }t  t |¡t |¡gt |¡t |¡ gg¡}t ||¡} |rÈ| d| dd¡7} t | ƒi} dD]F} tdd | d } |  | j ¡} | | | <| jjd kst‚| jd ksØt‚qØt| d | ddddS)z2Test FastICA is consistent between whiten_solvers.r(rrr%r=r>r@)rÚeighr)©r.r-Z whiten_solverrKrLr•rgê-™—q=rIN)r rrrPrQrrRrSrTrrZrandr{rUrVrrrr\r2r_rr)r;rbrrcrdrerfrgrhriZoutsZsolverrnrorrrÚ%test_fastica_simple_different_solvers˜s*  0  r—c Cs\tj |¡}| dd¡}||j}tdddd}d}tjt|d|  |¡W5QRXd S) z:Test FastICA eigh solver raises warning for low-rank data.rrrr)r•r–z$There are some small singular valuesrqN) r rrrrrrOrsrtr1)rbrÚAr6rnÚmsgrrrÚ"test_fastica_eigh_low_rank_warning»s   rš)r )%Ú__doc__rWrMr€Únumpyr rOZscipyrZsklearn.decompositionrrrZsklearn.decomposition._fasticarZsklearn.exceptionsrZsklearn.utils._testingrrr#r7r:ÚmarkZ parametrizerprwr|r}r…ÚfilterwarningsrˆrŒrŽr”r—ršrrrrÚsT        k  -2 úþ  "