Dokumentasi menyatakan bahwa R gbm dengan distribusi = "adaboost" dapat digunakan untuk masalah klasifikasi 0-1. Pertimbangkan fragmen kode berikut:
gbm_algorithm <- gbm(y ~ ., data = train_dataset, distribution = "adaboost", n.trees = 5000)
gbm_predicted <- predict(gbm_algorithm, test_dataset, n.trees = 5000)
Itu dapat ditemukan di dokumentasi yang memprediksi.gbm
Mengembalikan vektor prediksi. Secara default prediksi berada pada skala f (x).
Namun skala tertentu tidak jelas untuk kasus distribusi = "adaboost".
Adakah yang bisa membantu dengan interpretasi nilai return predict.gbm dan memberikan gagasan konversi ke output 0-1?
Jawaban:
Metode adaboost memberikan prediksi pada skala logit. Anda dapat mengubahnya menjadi output 0-1:
perhatikan 2 * di dalam logis
sumber
Anda juga dapat langsung mendapatkan probabilitas dari
predict.gbm
fungsi;sumber
Fungsi tautan adaboost dijelaskan di sini . Contoh ini memberikan deskripsi terperinci tentang perhitungan:
sumber