sklearn.utils.extmath.safe_sparse_dot¶

sklearn.utils.extmath.safe_sparse_dot(a, b, *, dense_output=False)[source]¶

Dot product that handle the sparse matrix case correctly

Parameters

aarray or sparse matrix

barray or sparse matrix

dense_outputboolean, (default=False)

When False, a and b both being sparse will yield sparse output. When True, output will always be a dense array.

Returns

dot_productarray or sparse matrix

sparse if a and b are sparse and dense_output=False.