Я пытаюсь классифицировать свои данные в одну из 3 категорий. Я использую логистическую регрессию sklearns, которая принимает в качестве входных данных слово для проверки векторизованной формы моих тестовых данных. Метод прогнозирования возвращает прогноз, но он не дает числа о вероятности или уверенности, что модель в классификации верна? Может кто-нибудь сказать мне, как я могу получить эту информацию?
Если вы присмотритесь к документации , есть другая функция, predict_proba, для получения вероятностей.
predict_proba
Так что вам просто нужно сделать:
clf = LogisticRegression() clf.predict_proba(X_input)