Предсказанные вероятности классов для обоих классов в R - PullRequest
0 голосов
/ 19 мая 2018

У меня есть проблема двоичной классификации, и я хотел бы получить предсказанные вероятности классов для обоих классов.Примерно так ...

0 0.25 0.15
0  0.2 0.19
1 0.19 0.25

Каждая строка содержит фактический класс элемента, прогнозируемую вероятность членства в классе 0 и прогнозируемую вероятность членства в классе 1. Я мог бы получить прогнозируемую вероятностьдля класса 1 с использованием функции прогнозирования

predict(mod,newx=ValidX,type="response")

Пожалуйста, дайте совет, как получить вероятности для класса 0.Заранее спасибо.

Ответы [ 2 ]

0 голосов
/ 21 мая 2018

У меня нет опыта работы с R, но для меня это больше похоже на вопрос общего машинного обучения.Что касается вашей проблемы (т. Е. Двоичной классификации с одним выходом), вероятно, вам следует рассмотреть ответ , предоставленный tyumru .

При этом некоторые алгоритмы машинного обучения будут выводитьвероятность для каждого класса из коробки.Например, если вы используете нейронную сеть с функцией активации softmax на выходном слое, вы получите вероятность для каждого класса.

0 голосов
/ 19 мая 2018

Поскольку это двоичная классификация, Prob (C2) = 1 - Prob (C1)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...