Scikit Узнайте значение выходных данных ** Предсказание ** и ** Предсказание. ** - PullRequest
1 голос
/ 21 января 2020

При использовании Scikit Learn для Logisti c Регрессия существуют коды:

print(sentiment_model.predict_proba(test_features[:2]))
print(sentiment_model.predict(test_features[:2]))

Вывод:

[[1.20150248e-04 9.99879850e-01]
 [9.99986356e-01 1.36436948e-05]]
[1 0]

Я изо всех сил пытаюсь понять смысл вывода Предсказание_проба и Предсказание. Спасибо за ваше время.

1 Ответ

1 голос
/ 21 января 2020

В Scikit-Learn predict возвращает прогноз (0 или 1, False или True), а predict_proba возвращает оценку достоверности для прогноза. Выходные данные predict_proba представляют собой массив, в котором первый элемент является оценкой вероятности того, что экземпляр является 0 классом (или False классом), а второй элемент является оценкой вероятности того, что экземпляр является 1 класс (или True класс). Таким образом, строки в выводе predict_proba складываются в единицу.

В вашем коде первый экземпляр - это класс 1 (независимо от того, что было закодировано, чтобы означать в вашем приложении), с вероятностью 0.999879859. Во втором случае, однако, вероятность того, что это будет класс 1, мала (0.00001364...). Следовательно, это класс 0 с вероятностью 0.99998635).

...