{KgN~dZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z mZmZddlmZGd d e e e e Zy) z)Principal Component Analysis Base Classes)ABCMetaabstractmethodN)linalg) BaseEstimatorClassNamePrefixFeaturesOutMixinTransformerMixin)_add_to_diagonaldevice get_namespace)check_is_fittedcReZdZdZdZdZed dZdZd dZ dZ e d Z y) _BasePCAzwBase class for PCA methods. Warning: This class should not be used directly. Use derived classes instead. c t|j\}}|j}|j}|jr)||j |ddt j fz}||jz }|j||jkD||jdt|}|j|z|z}t||j||S)asCompute data covariance with the generative model. ``cov = components_.T * S**2 * components_ + sigma2 * eye(n_features)`` where S**2 contains the explained variances, and sigma2 contains the noise variances. Returns ------- cov : array of shape=(n_features, n_features) Estimated covariance of data. Nr ) r components_explained_variance_whitensqrtnpnewaxisnoise_variance_whereasarrayr Tr )selfxp_rexp_var exp_var_diffcovs _/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/sklearn/decomposition/_base.pyget_covariancez_BasePCA.get_covariancesd../A&& ** ;;%2:: 0F(GGK!5!55 xx d** *  JJs6'?J 3 }}|+{:d22B7 c lt|j\}}|jjd}|jdk(r|j ||j z S|r|j j}nt j}|j dk(r||jS|j}|j}|jr)||j|ddtjfz}||j z }|j||j kD||jdt!|}||j"z|j z }t%|d|z ||j"||z|z}||j dz z}t%|d|j z ||S)a8Compute data precision matrix with the generative model. Equals the inverse of the covariance but computed with the matrix inversion lemma for efficiency. Returns ------- precision : array, shape=(n_features, n_features) Estimated precision of data. rrNrg?r)r rshape n_components_eyerrinvr$rrrrrrrr rr ) rris_array_api_compliant n_features linalg_invrr r! precisions r# get_precisionz_BasePCA.get_precision:s&343C3C%D" "%%++A.     "66*%(<(<< < !JJ   3 &d1134 4&& ** ;;%2:: 0F(GGK!5!55 xx d** *  JJs6'?J 3  +--/$2F2FF C,$6;MMJy$99KG t++Q.// C$*>*>$>Cr%Ncy)aPlaceholder for fit. Subclasses should implement this method! Fit the model with X. Parameters ---------- X : array-like of shape (n_samples, n_features) Training data, where `n_samples` is the number of samples and `n_features` is the number of features. Returns ------- self : object Returns the instance itself. N)rXys r#fitz _BasePCA.fitgsr%ct||j|j\}}t||j ||j |j gdd}|j||dS)aApply dimensionality reduction to X. X is projected on the first principal components previously extracted from a training set. Parameters ---------- X : {array-like, sparse matrix} of shape (n_samples, n_features) New data, where `n_samples` is the number of samples and `n_features` is the number of features. Returns ------- X_new : array-like of shape (n_samples, n_components) Projection of X in the first principal components, where `n_samples` is the number of samples and `n_components` is the number of the components. )csrcscF)dtype accept_sparsereset)r x_is_centered)r rrr _validate_datafloat64float32 _transform)rr3rrs r# transformz_BasePCA.transformysm$a!1!143K3KLA    bjj"**-^SX  qRu==r%cZ||jjz}|s6||j|jd|jjzz}|jrM|j |j }|j|jj}||||k<||z}|S)N)r') rrreshapemean_rrrfinfor9eps)rr3rr< X_transformedscale min_scales r#r@z_BasePCA._transformsD,,...  RZZ G6(5@))r%r) metaclass)rSabcrrnumpyrscipyrbaserrr utils._array_apir r r utils.validationr rr2r%r#r\s9/(SSFF.v)#%5}PWv)r%