Q3jx d Z ddlZddlZddlmZmZ ddlmZ ddl m
Z
m Z ddl m
Z
ddgZ ej d
d Zej" j% d dg ej" j% d
d d Zej" j% dd d Zej" j% dg d d Zej" j% dd d Zej" j% de d Zd Zej" j% d
d ej" j% dd d g ej" j% d e d Zej" j% d
d ej" j% d e d Zej" j% d e d Zej" j% d e d! Z d" Z ej" j% d#g d$ ej" j% d
d d% Z y)&z Test truncated SVD transformer. N)PCA TruncatedSVD)check_random_state)assert_allcloseassert_array_less)_sparse_random_arrayarpack
randomizedmodule)scopec t d } t ddd| }dt j |j z |j d d |S )N* )< 7 皙?csr)densityformatrng )r r nplogdata)r Xs [/DATA/.local/lib/python3.12/site-packages/sklearn/decomposition/tests/test_truncated_svd.pyX_sparser sA R
C Xs5cJABFF166N"AFF1I H solverkind)densesparsec |dk( r| n| j }t dd }t d|dd }|j | d d d df }|j | d d d df }t ||d
t j
|j }t j
|j } t |d d | d d d
t |d d | d d d
y )Nr! r algorithmr d )r% random_state
n_oversamples gMb`?rtol gMbP?{Gz?atol)toarrayr
fit_transformr r abs components_)
r r r r svd_asvdXaXrcomp_acomps
r test_solversr: s H$(*:*:*! %%%r c ^ |dk( r| n| j }t d|d j | }t d|d j | }t |j |j d d d |j j |j j kD sJ y )Nr! r< )r% n_iterrf g{Gzt?r* )r0 r rA r rl rm )r r r r svd_10svd_20s r (test_explained_variance_components_10_20rx t s H$(*:*:*
>q
AF
"r
:
>
>q
AF (((("-
((,,.1Q1Q1U1U1WWWr c t j j | }d\ }}|j || }t d| | j | }|j
| }t t j |j dz t j j |d dz d t |j t j t j |dz d d y )
N)r& P r; r% r' @fror- r* r rj )
r random RandomStaterandnr rA rU r rm singular_values_linalgnormsqrt)r global_random_seedr rV rB r pcaX_pcas r test_singular_values_consistencyr s
))
2
3C #Iz )Z(A
Ac
J
N
Nq
QC
MM! E
s##S()
ue$+
bggbffUCZa&@ Ar c t j j | }d}d}|j || }t d| | }|j | }|t j t j |dz d z }|d d dfxx dz cc< |d d d fxx d
z cc< t j ||j }|j | t |j g d d
y )Nr& n r| r} r rj A`" @r X9v@)r r ri g+=r* )
r r r r r r1 r rm dotr3 rA r r ) r r r rV rB r r r X_hat_pcas r test_singular_values_expectedr s
))
2
3CIJ )Z(A
Ac
JC a E RWWRVVE3JQ/
00E !Q$K5K !Q$K5Kucoo.IGGIC((*=EJr c l | j }||j d z
}t dd }t d ddi|}t d ddi|}|j | }|j | }t
||d
t
|j dd t
|j |j y )
Nr rj r< r )r; r' r% r
svd_solverg& .>r* r. ) r0 meandictr r r1 r mean_r3 )r X_denseX_cparamsr5 r Xt_svdXt_pcas r test_truncated_svd_eq_pcar s G
GLLaL(
(C
r
3F
4
4V
4C
,
,V
,C
s
#F
s
#FFF.CIIqt,COOS__5r zalgorithm, tol, normalizer))r
rh auto)r
rh QR)r
rh LU)r
rh none)r gư>r )r rh r c |dk( r| n| j }t ddd||| }|j | }|j | j | }t || y )Nr! r )r; ru r' r% power_iteration_normalizertol)r0 r r1 rA rU r ) r r% r r
normalizerr r5 X_transformed_1X_transformed_2s r test_fit_transformr sk H$(*:*:*r s] %
3 , E 4 & h
L>2!45
5 6 3
5 )9:@ ;@ EF+ G+ -D .D: !45 "b2;/ 0 3 6* !45;/ 0 6$ ;/ 0, ;/K 0K(6(
!45
6 6
6r