способ настройки порога для метода прогнозирования в sklearn.ensemble.GradientBoostingClassifier - PullRequest
0 голосов
/ 13 октября 2018

Я использую классификатор повышения градиента, реализованный в scikit-learn (sklearn.ensemble.GradientBoostingClassifier) ​​для проблемы двоичной классификации.Хотя метод прогнозирования предоставляется по умолчанию, в официальных документах не было никаких подсказок:

  1. как определен порог по умолчанию,

  2. лиесть индивидуальный способ изменения этого порога во время настройки моделей

Любые подсказки и советы будут приветствоваться

Большое спасибо

1 Ответ

0 голосов
/ 13 октября 2018

Что касается 1, я думаю, что это .5

Что касается 2, когда я использовал библиотеки sklearn, я использовал предикат_proba, который затем вы можете пороговать самостоятельно
http://scikit -learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html#sklearn.ensemble.GradientBoostingClassifier.predict_proba
(таким образом, вы также можете проверить, какой порог по умолчанию сравнивается)

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