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

Question

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

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

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

Answer 1

Nevermind, я нашел решение .:

predict_proba (X) Возвращает оценку вероятности для испытаний вектора X.