]jsaSrSSKrSSKrSSKJr SSKJr SSK J r J r SSK J r SSK Jr SSK Jr SSK Jr SS KJr S rS rS rS rSSjrSS.SjrSrSrSrSrSr Sr!SSjr"S Sjr#Sr$Sr%Sr&Sr'S!Sjr(g)"zBA collection of utilities to work with sparse matrices and arrays.N)LinearOperator)_sparse_min_max_sparse_nan_min_max)csc_mean_variance_axis0)csr_matmul_csr_to_dense)csr_mean_variance_axis0)incr_mean_variance_axis0)_check_sample_weightc[R"U5(a URO [U5nSU-n[ U5e)z2Raises a TypeError if X is not a CSR or CSC matrixz,Expected a CSR or CSC sparse matrix, got %s.)spissparseformattype TypeError)X input_typeerrs V/home/mande/repo/quber/.venv/lib/python3.13/site-packages/sklearn/utils/sparsefuncs.py_raise_typeerrorrs1[[^^aJ 8: EC C.c,US;a[SU-5eg)N)rz8Unknown axis value: %d. Use 0 for rows, or 1 for columns) ValueErroraxiss r_raise_error_wrong_axisr#s$ 6 F M  rcURSURS:XdeU=RURURSS9-slg)aInplace column scaling of a CSR matrix. Scale each feature of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to normalize using the variance of the features. It should be of CSR format. scale : ndarray of shape (n_features,), dtype={np.float32, np.float64} Array of precomputed feature-wise values to use for scaling. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 3, 4, 4, 4]) >>> indices = np.array([0, 1, 2, 2]) >>> data = np.array([8, 1, 2, 5]) >>> scale = np.array([2, 3, 2]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 1, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) >>> sparsefuncs.inplace_csr_column_scale(csr, scale) >>> csr.todense() matrix([[16, 3, 4], [ 0, 0, 10], [ 0, 0, 0], [ 0, 0, 0]]) rrclip)modeN)shapedatatakeindicesrscales rinplace_csr_column_scaler&*sBJ ;;q>QWWQZ '' 'FFejjj00FrcURSURS:XdeU=R[R"U[R"UR 55-slg)aInplace row scaling of a CSR matrix. Scale each sample of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to be scaled. It should be of CSR format. scale : ndarray of float of shape (n_samples,) Array of precomputed sample-wise values to use for scaling. rN)r r!nprepeatdiffindptrr$s rinplace_csr_row_scaler,SsH ;;q>QWWQZ '' 'FFbiirwwqxx011Frcl[U5 [R"U5(a4URS:Xa$US:Xa [ XUS9$[ UR X#S9$[R"U5(a4URS:Xa$US:Xa [ XUS9$[ UR X#S9$[U5 g)aCompute mean and variance along an axis on a CSR or CSC matrix. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Input data. It can be of CSR or CSC format. axis : {0, 1} Axis along which the axis should be computed. weights : ndarray of shape (n_samples,) or (n_features,), default=None If axis is set to 0 shape is (n_samples,) or if axis is set to 1 shape is (n_features,). If it is set to None, then samples are equally weighted. .. versionadded:: 0.24 return_sum_weights : bool, default=False If True, returns the sum of weights seen for each feature if `axis=0` or each sample if `axis=1`. .. versionadded:: 0.24 Returns ------- means : ndarray of shape (n_features,), dtype=floating Feature-wise means. variances : ndarray of shape (n_features,), dtype=floating Feature-wise variances. sum_weights : ndarray of shape (n_features,), dtype=floating Returned if `return_sum_weights` is `True`. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 3, 4, 4, 4]) >>> indices = np.array([0, 1, 2, 2]) >>> data = np.array([8, 1, 2, 5]) >>> scale = np.array([2, 3, 2]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 1, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) >>> sparsefuncs.mean_variance_axis(csr, axis=0) (array([2. , 0.25, 1.75]), array([12. , 0.1875, 4.1875])) csrr)weightsreturn_sum_weightscscN)rr r r_csr_mean_var_axis0_csc_mean_var_axis0Tr)rrr/r0s rmean_variance_axisr5eslD! {{1~~!((e+ 19&7I 'W  QAHH- 19&7I 'W  r)r/c[U5 [R"U5(aURS;d [ U5 [ R "U5S:Xa)[ R"URXBRS9n[ R "U5[ R "U5s=:Xa[ R "U5:Xd O [S5eUS:Xa[[ R "U5URS:wa3[SURSS[ R "U5S35eOZ[ R "U5URS:wa3[S URSS[ R "U5S35eUS:Xa UROUnUb[XPURS9n[XX4US 9$) a Compute incremental mean and variance along an axis on a CSR or CSC matrix. last_mean, last_var are the statistics computed at the last step by this function. Both must be initialized to 0-arrays of the proper size, i.e. the number of features in X. last_n is the number of samples encountered until now. Parameters ---------- X : CSR or CSC sparse matrix of shape (n_samples, n_features) Input data. axis : {0, 1} Axis along which the axis should be computed. last_mean : ndarray of shape (n_features,) or (n_samples,), dtype=floating Array of means to update with the new data X. Should be of shape (n_features,) if axis=0 or (n_samples,) if axis=1. last_var : ndarray of shape (n_features,) or (n_samples,), dtype=floating Array of variances to update with the new data X. Should be of shape (n_features,) if axis=0 or (n_samples,) if axis=1. last_n : float or ndarray of shape (n_features,) or (n_samples,), dtype=floating Sum of the weights seen so far, excluding the current weights If not float, it should be of shape (n_features,) if axis=0 or (n_samples,) if axis=1. If float it corresponds to having same weights for all samples (or features). weights : ndarray of shape (n_samples,) or (n_features,), default=None If axis is set to 0 shape is (n_samples,) or if axis is set to 1 shape is (n_features,). If it is set to None, then samples are equally weighted. .. versionadded:: 0.24 Returns ------- means : ndarray of shape (n_features,) or (n_samples,), dtype=floating Updated feature-wise means if axis = 0 or sample-wise means if axis = 1. variances : ndarray of shape (n_features,) or (n_samples,), dtype=floating Updated feature-wise variances if axis = 0 or sample-wise variances if axis = 1. n : ndarray of shape (n_features,) or (n_samples,), dtype=integral Updated number of seen samples per feature if axis=0 or number of seen features per sample if axis=1. If weights is not None, n is a sum of the weights of the seen samples or features instead of the actual number of seen samples or features. Notes ----- NaNs are ignored in the algorithm. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 3, 4, 4, 4]) >>> indices = np.array([0, 1, 2, 2]) >>> data = np.array([8, 1, 2, 5]) >>> scale = np.array([2, 3, 2]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 1, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) >>> sparsefuncs.incr_mean_variance_axis( ... csr, axis=0, last_mean=np.zeros(3), last_var=np.zeros(3), last_n=2 ... ) (array([1.33, 0.167, 1.17]), array([8.88, 0.139, 3.47]), array([6., 6., 6.])) r1r.rdtypez8last_mean, last_var, last_n do not have the same shapes.rzHIf axis=1, then last_mean, last_n, last_var should be of size n_samples z (Got z).zIIf axis=0, then last_mean, last_n, last_var should be of size n_features ) last_meanlast_varlast_nr/)rr r rrr(sizefullr r9rr4r _incr_mean_var_axis0)rrr:r;r<r/s rincr_mean_variance_axisr@ssbD! KKNNqxx>9 wwv!&H GGI "''("3 Frwwv FSTT qy 779  +""#''!*VBGGI4F3GrK  , 779  +##$771:,fRWWY5G4HL  qyaA&wA  ' rc [R"U5(a'URS:Xa[URU5 g[R"U5(aURS:Xa [ X5 g[ U5 g)aInplace column scaling of a CSC/CSR matrix. Scale each feature of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to normalize using the variance of the features. It should be of CSC or CSR format. scale : ndarray of shape (n_features,), dtype={np.float32, np.float64} Array of precomputed feature-wise values to use for scaling. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 3, 4, 4, 4]) >>> indices = np.array([0, 1, 2, 2]) >>> data = np.array([8, 1, 2, 5]) >>> scale = np.array([2, 3, 2]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 1, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) >>> sparsefuncs.inplace_column_scale(csr, scale) >>> csr.todense() matrix([[16, 3, 4], [ 0, 0, 10], [ 0, 0, 0], [ 0, 0, 0]]) r1r.N)r r rr,r4r&rr$s rinplace_column_scalerB&sSJ {{1~~!((e+acc5) QAHH- *rc [R"U5(a'URS:Xa[URU5 g[R"U5(aURS:Xa [ X5 g[ U5 g)aInplace row scaling of a CSR or CSC matrix. Scale each row of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to be scaled. It should be of CSR or CSC format. scale : ndarray of shape (n_features,), dtype={np.float32, np.float64} Array of precomputed sample-wise values to use for scaling. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 2, 3, 4, 5]) >>> indices = np.array([0, 1, 2, 3, 3]) >>> data = np.array([8, 1, 2, 5, 6]) >>> scale = np.array([2, 3, 4, 5]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 1, 0, 0], [0, 0, 2, 0], [0, 0, 0, 5], [0, 0, 0, 6]]) >>> sparsefuncs.inplace_row_scale(csr, scale) >>> csr.todense() matrix([[16, 2, 0, 0], [ 0, 0, 6, 0], [ 0, 0, 0, 20], [ 0, 0, 0, 30]]) r1r.N)r r rr&r4r,rr$s rinplace_row_scalerDSsSH {{1~~!((e+ e, QAHH-a'rc8X4H-n[U[R5(dM$[S5e US:aXRS- nUS:aX RS- nUR U:HnXR UR U:H'X R U'g)a#Swap two rows of a CSC matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two rows are to be swapped. It should be of CSC format. m : int Index of the row of X to be swapped. n : int Index of the row of X to be swapped. m and n should be valid integersrN) isinstancer(ndarrayrr r#)rmntm_masks rinplace_swap_row_cscrMsV a $ $>? ? 1u WWQZ1u WWQZ YY!^F !IIaii1nIIfrc 0X4H-n[U[R5(dM$[S5e US:aXRS- nUS:aX RS- nX:aX!p!UR nXAnXAS-nXBnXBS-nXe- n X- n X:waBUR US-U===X- - sss&XZ-UR US-'X- UR U'[R "URSUURXxURXgURXVURUS/5Ul[R "URSUURXxURXgURXVURUS/5Ulg)a#Swap two rows of a CSR matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two rows are to be swapped. It should be of CSR format. m : int Index of the row of X to be swapped. n : int Index of the row of X to be swapped. rFrrN) rGr(rHrr r+ concatenater#r!) rrIrJrKr+m_startm_stopn_startn_stopnz_mnz_ns rinplace_swap_row_csrrWsV a $ $>? ? 1u WWQZ1u WWQZ u1 XXFiG E]FiG E]F  D  D | Qt{*!.Qm  IIhw  IIg % IIf % IIg % IIfg   AI^^ FF8G  FF7 " FF6 " FF7 " FF67O  AFrc[R"U5(aURS:Xa [XU5 g[R"U5(aURS:Xa [ XU5 g[ U5 g)a' Swap two rows of a CSC/CSR matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two rows are to be swapped. It should be of CSR or CSC format. m : int Index of the row of X to be swapped. n : int Index of the row of X to be swapped. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 2, 3, 3, 3]) >>> indices = np.array([0, 2, 2]) >>> data = np.array([8, 2, 5]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 0, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) >>> sparsefuncs.inplace_swap_row(csr, 0, 1) >>> csr.todense() matrix([[0, 0, 5], [8, 0, 2], [0, 0, 0], [0, 0, 0]]) r1r.N)r r rrMrWrrrIrJs rinplace_swap_rowrZsQJ {{1~~!((e+Q1% QAHH-Q1%rcVUS:aXRS- nUS:aX RS- n[R"U5(aURS:Xa [ XU5 g[R"U5(aURS:Xa [ XU5 g[ U5 g)a6 Swap two columns of a CSC/CSR matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two columns are to be swapped. It should be of CSR or CSC format. m : int Index of the column of X to be swapped. n : int Index of the column of X to be swapped. Examples -------- >>> from sklearn.utils import sparsefuncs >>> from scipy import sparse >>> import numpy as np >>> indptr = np.array([0, 2, 3, 3, 3]) >>> indices = np.array([0, 2, 2]) >>> data = np.array([8, 2, 5]) >>> csr = sparse.csr_matrix((data, indices, indptr)) >>> csr.todense() matrix([[8, 0, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) >>> sparsefuncs.inplace_swap_column(csr, 0, 1) >>> csr.todense() matrix([[0, 8, 2], [0, 0, 5], [0, 0, 0], [0, 0, 0]]) rrr1r.N)r r r rrWrMrrYs rinplace_swap_columnr\s}J 1u WWQZ1u WWQZ {{1~~!((e+Q1% QAHH-Q1%rc[R"U5(a)URS;aU(a [XS9$[ XS9$[ U5 g)aYCompute minimum and maximum along an axis on a CSR or CSC matrix. Optionally ignore NaN values. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Input data. It should be of CSR or CSC format. axis : {0, 1} Axis along which the axis should be computed. ignore_nan : bool, default=False Ignore or passing through NaN values. .. versionadded:: 0.20 Returns ------- mins : ndarray of shape (n_features,), dtype={np.float32, np.float64} Feature-wise minima. maxs : ndarray of shape (n_features,), dtype={np.float32, np.float64} Feature-wise maxima. )r.r1rN)r r rrrr)rr ignore_nans r min_max_axisr_9s=6 {{1~~!((n4 &q4 4"10 0rcUS:XaSnO=US:XaSnO4URS:wa$[SRUR55eUcDUc UR$[R"[R "UR 5U5$US:Xa8[R "UR 5nUcURS5$X2-$US:XaUc,[R"URURSS9$[R"U[R "UR 55n[R"URURSUS 9$[S RU55e) aA variant of X.getnnz() with extension to weighting on axis 0. Useful in efficiently calculating multilabel metrics. Parameters ---------- X : sparse matrix of shape (n_samples, n_labels) Input data. It should be of CSR format. axis : {0, 1}, default=None The axis on which the data is aggregated. sample_weight : array-like of shape (n_samples,), default=None Weight for each row of X. Returns ------- nnz : int, float, ndarray of shape (n_samples,) or ndarray of shape (n_features,) Number of non-zero values in the array along a given axis. Otherwise, the total number of non-zero values in the array is returned. rrr.z#Expected CSR sparse format, got {0}intp) minlength)rdr/zUnsupported axis: {0}) rrnnzr(dotr*r+astypebincountr#r r)r)rr sample_weightoutr/s r count_nonzerork]s+, rz  U =DDQXXNOO  |  55L66"''!((+]; ; ggahh  ::f% %""   ;;qyyAGGAJ? ?ii rwwqxx/@AG;;qyyAGGAJP P077=>>rc [U5U-nU(d[R$[R"US:5n[ US5upEUR 5 U(a [ X@X15$[ US- XU5[ X@X15-S- $)zCompute the median of data with n_zeros additional zeros. This function is used to support sparse matrices; it modifies data in-place. rrOrg@)lenr(nanrkdivmodsort_get_elem_at_rank)r!n_zerosn_elems n_negativemiddleis_odds r _get_medianrws $i'!G vv !!$(+JGQ'NFIIK  zCC &1*d@ F* > ?   rc4X:aX$X- U:agXU- $)z@Find the value in data augmented with n_zeros for the given rankr)rankr!rtrrs rrqrqs+ z 7" w rc[R"U5(aURS:Xd[SUR-5eURnUR up#[ R"U5n[[R"U55HEunupg[ R"URXg5nX(R- n [X5XE'MG U$)aFind the median across axis 0 of a CSC matrix. It is equivalent to doing np.median(X, axis=0). Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Input data. It should be of CSC format. Returns ------- median : ndarray of shape (n_features,) Median. r1z%Expected matrix of CSC format, got %s)r r rrr+r r(zeros enumerate itertoolspairwisecopyr!r=rw) rr+ n_samples n_featuresmedianf_indstartendr!nzs rcsc_median_axis_0rs KKNNqxx50?!((JKK XXFGGI XXj !F(););F)CD|wwqvve()  "#D- E Mrc ^^^TSSS24mTRm[UU4SjUU4SjUU4SjUU4SjTRTRS9$)aCreate an implicitly offset linear operator. This is used by PCA on sparse data to avoid densifying the whole data matrix. Params ------ X : sparse matrix of shape (n_samples, n_features) offset : ndarray of shape (n_features,) Returns ------- centered : LinearOperator Nc>TU-TU-- $Nryxroffsets r)_implicit_column_offset..Q!+rc>TU-TU-- $rryrs rrrrrc6>TU-TUR5-- $r)sumrXTrs rrrs"q&FQUUW$45rcV>TU-TRURSS9SSS24-- $)Nrr)r4rrs rrrs*"q&688aeeemD!G.D#DDr)matvecmatmatrmatvecrmatmatr9r )r4rr9r )rrrs``@r_implicit_column_offsetrsBD!G_F B ++5Dgggg  rc r[R"U5(a URS;aURS:Xd [ S5e[R"U5(a URS;aURS:Xd [ S5eUR SUR S:wa.SUR SSUR SS 3n[ U5eUR upEUR SnUR UR :wd.UR [R[R4;a S n[ U5eUc+[R"XF4URR S 9nO`UR SU:wdUR SU:wa [ S 5eUR URR :wa [ S 5eSnURS:XaIURS:Xa(SnURURURp!nXdpdO1UR5nO URS:XaUR5n[URURUR URURUR X$XV5 U(a URnU$)aCompute A @ B for sparse and 2-dim A and B while returning an ndarray. Parameters ---------- A : sparse matrix of shape (n1, n2) and format CSC or CSR Left-side input matrix. B : sparse matrix of shape (n2, n3) and format CSC or CSR Right-side input matrix. out : ndarray of shape (n1, n3) or None Optional ndarray into which the result is written. Returns ------- out An ndarray, new created if out=None. r7rOz2Input 'A' must be a sparse 2-dim CSC or CSR array.z2Input 'B' must be a sparse 2-dim CSC or CSR array.rrz1Shapes must fulfil A.shape[1] == B.shape[0], got z == .zBDtype of A and B must be the same, either both float32 or float64.r8z3Shape of out must be ({n1}, {n3}), got {out.shape}.z)Dtype of out must match that of input A..Fr1T)r r rndimrr r9r(float32float64emptyr!r4tocsrrr#r+)ABrjmsgn1n2n3 transpose_outs rsparse_matmul_to_densers" KKNNqxx>9affkMNN KKNNqxx>9affkMNNwwqzQWWQZ 771:,d1771:,a 1 o WWFB Bww!''QWWRZZ,DDRo {hhxqvv||4 99Q<2 1!3RS S 99 $HI IMxx5 88u  MQSS#%%#A A U  GGI  188QVVQYY#2ee Jr)NF)F)NNr))__doc__r~numpyr( scipy.sparsesparser scipy.sparse.linalgrsklearn.utils.fixesrrsklearn.utils.sparsefuncs_fastrr3rrr2r r?sklearn.utils.validationr rrr&r,r5r@rBrDrMrWrZr\r_rkrwrqrrrryrrrsH .D: &1R2$K\NRpf*Z)X:<~*Z.b!H3?l , >6<r