Документация утверждает, что R gbm с distribution = "adaboost" может использоваться для задачи классификации 0-1. Рассмотрим следующий фрагмент кода:
gbm_algorithm <- gbm(y ~ ., data = train_dataset, distribution = "adaboost", n.trees = 5000)
gbm_predicted <- predict(gbm_algorithm, test_dataset, n.trees = 5000)
Это можно найти в документации, которая прогнозирует.gbm
Возвращает вектор прогнозов. По умолчанию прогнозы находятся в масштабе f (x).
Однако конкретный масштаб неясен для случая distribution = "adaboost".
Может ли кто-нибудь помочь с интерпретацией предиктивных значений предиката.gtm и дать представление о преобразовании в вывод 0-1?
Ответы:
Метод adaboost дает прогнозы в логитном масштабе. Вы можете преобразовать его в вывод 0-1:
обратите внимание на 2 * внутри логиса
источник
Вы также можете напрямую получить вероятности из
predict.gbm
функции;источник
Функция adaboost link описана здесь . Этот пример предоставляет подробное описание вычисления:
источник