o ?Hh> @s`dZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z Gd d d eZdS) zY Feature agglomeration. Base classes and functions for performing feature agglomeration. N)issparse)TransformerMixin)metadata_routing)"_deprecate_Xt_in_inverse_transform)check_is_fitted validate_datac@s2eZdZdZdejiZddZd ddddZdS) AgglomerationTransformzH A class for feature agglomeration via the transform interface. Xtcsttddjtjkr2ts2tjjd}t fddt |D}|Sfddt jD}t |j }|S)a Transform a new matrix using the built clustering. Parameters ---------- X : array-like of shape (n_samples, n_features) or (n_samples, n_samples) A M by N array of M observations in N dimensions or a length M array of M one-dimensional observations. Returns ------- Y : ndarray of shape (n_samples, n_clusters) or (n_clusters,) The pooled values for each feature cluster. F)resetrc s*g|]}tj|ddfqSN)npbincountlabels_).0iXselfsizef/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sklearn/cluster/_feature_agglomeration.py 8s*z4AgglomerationTransform.transform..cs,g|]}jddj|kfddqS)N)axis) pooling_funcr)rl)rrrrr;s) rrrr meanrrrshapearrayrangeuniqueT)rr n_samplesnXrrr transform s    z AgglomerationTransform.transformN)r cCs2t||}t|tj|jdd\}}|d|fS)a Inverse the transformation and return a vector of size `n_features`. Parameters ---------- X : array-like of shape (n_samples, n_clusters) or (n_clusters,) The values to be assigned to each cluster of samples. Xt : array-like of shape (n_samples, n_clusters) or (n_clusters,) The values to be assigned to each cluster of samples. .. deprecated:: 1.5 `Xt` was deprecated in 1.5 and will be removed in 1.7. Use `X` instead. Returns ------- X : ndarray of shape (n_samples, n_features) or (n_features,) A vector of size `n_samples` with the values of `Xred` assigned to each of the cluster of samples. T)return_inverse.)rrr r!r)rrr unilinverserrrinverse_transformBs  z(AgglomerationTransform.inverse_transformr ) __name__ __module__ __qualname____doc__rUNUSED<_AgglomerationTransform__metadata_request__inverse_transformr%r)rrrrr s  "r )r-numpyr scipy.sparserbaserutilsrutils.deprecationrutils.validationrrr rrrrs