safe_sparse_dot

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

正確處理稀疏矩陣情況的點積。

參數:

a{ndarray, 稀疏矩陣}

b{ndarray, 稀疏矩陣}

dense_outputbool, 預設值=False

當 False 時，如果 a 和 b 都是稀疏矩陣，則會產生稀疏輸出。當 True 時，輸出將永遠是密集陣列。

回傳值:

dot_product{ndarray, 稀疏矩陣}

如果 a 和 b 是稀疏矩陣且 dense_output=False，則為稀疏矩陣。

範例

>>> 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]])