safe_sparse_dot

sklearn.utils.extmath.safe_sparse_dot(a, b, *, dense_output=False)

Skalarprodukt, das den Fall von spärlichen Matrizen korrekt behandelt.

Parameter:

a{ndarray, sparse matrix}

b{ndarray, sparse matrix}

dense_outputbool, standardmäßig False

Wenn False, ergeben a und b beide sparse, was zu sparse Ausgabe führt. Wenn True, ist die Ausgabe immer ein dichtes Array.

Gibt zurück:

dot_product{ndarray, sparse matrix}

Sparse, wenn a und b sparse sind und dense_output=False.

Beispiele

>>> from scipy.sparse import csr_matrix
>>> from sklearn.utils.extmath import safe_sparse_dot
>>> X = csr_matrix([[1, 2], [3, 4], [5, 6]])
>>> dot_product = safe_sparse_dot(X, X.T)
>>> dot_product.toarray()
array([[ 5, 11, 17],
       [11, 25, 39],
       [17, 39, 61]])