Понимание результатов предсказания - PullRequest
0 голосов
/ 21 января 2020

Я использовал глубокую нейронную сеть для прогнозирования (которая может быть 1 или 0). Решенная проблема является проблемой классификации. Я использую predict_proba из sklearn, чтобы предсказать вероятности моего прогноза. Насколько мне известно, вывод predict_proba должен иметь 2 столбца в строке, обозначающих вероятность каждого класса. Я ожидаю что-то вроде этого:

[0.99467199 0.00532801]
[0.00223842 0.99776158]
[0.99389073 0.00610927]

Но я получаю:

[[6.92488432e-01]
 [5.56170940e-04]
 [7.24569559e-01]
 [9.99984384e-01]]

Как я могу интерпретировать эти результаты? Эти значения, соответствующие вероятности класса, были предсказаны?

1 Ответ

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

Это нормально, когда у вас есть только два класса, достаточно дать только один проба одного из них, потому что другой можно найти, вычтя пробу первого класса из 1.

...