Работает ли разреженная матрица с MultinomialNB?

Question

У меня есть векторы формы BoW (100000, 56000), и я хочу использовать MultinomialNB из scikit-learn для задачи классификации.

Принимает ли MultinomialNB разреженную матрицу для подгонки данных?

Я не могу преобразовать его в плотную матрицу toarray() из-за ошибки памяти. Если классификатор NB не принимает разреженную матрицу, есть ли альтернативы, которые я мог бы использовать для подгонки данных без преобразования их в плотную матрицу?

Answer 1

Из документации из MultinomialNB.fit (выделение добавлено):

fit(X, y, sample_weight=None)

Параметры:

X: {в виде массива, разреженная матрица }, shape = [n_samples, n_features]

Обучающие векторы, где n_samples - количество выборок, а n_features - количество объектов.