以下の予測関数は-ve値も提供しているため、確率にはなりません。
param <- list(max.depth = 5, eta = 0.01, objective="binary:logistic",subsample=0.9)
bst <- xgboost(param, data = x_mat, label = y_mat,nround = 3000)
pred_s <- predict(bst, x_mat_s2)
グーグルで試しましたpred_s <- predict(bst, x_mat_s2,type="response")
が、うまくいきませんでした。
質問
代わりに確率を予測する方法は?
使用した設定でデフォルトで確率を出力しませんか?つまり、pred_sを調べましたが、これらは確率ではないことは確かです。
—
kpb
負の値はありません。確率は
—
0〜1の
明らかな問題は見当たりません。(ただし、私はpythonラッパーに精通しています)。関数に追加
—
反転
outputmargin=F
してみましたpredict
か?何とか場合outputmargin
に設定されているT
、それはロジスティック変換前の値を返します。
Pythonのためには、コピーすることができます
—
アントンタラセンコ
predict_proba
から実装をsklearn
API:github.com/dmlc/xgboost/blob/master/python-package/xgboost/...