sklearn NB классификатор: как получить фактические вероятности отдельных выборок?

Question

Я делаю программу машинного обучения, которая классифицирует слова в одну из следующих категорий: Аппаратное обеспечение, Программное обеспечение, None_of_these. Я использую Multinomial Naive Bayes классификатор из sklearn.

Функция предиката () дает мне прогноз каждого слова, однако я не вижу фактической вероятности (с плавающей запятой в диапазоне от 0 до 1,0), что слово соответствует предсказанной категории. Я не нашел этого на сайте sklearn.

Есть ли функция, которая дает мне вероятность каждого образца?

Answer 1

Неважно, я нашел решение.

Предсказание (X) Возвращает оценки вероятности для тестового вектора X.