Poniższa funkcja przewidywania podaje również wartości -ve, więc nie może to być prawdopodobieństwo.
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)
I google i próbowałem, pred_s <- predict(bst, x_mat_s2,type="response")
ale to nie działało.
Pytanie
Jak zamiast tego przewidzieć prawdopodobieństwa?
outputmargin=F
do predict
funkcji? Jeśli w jakiś sposób outputmargin
jest ustawiony na T
, zwróci wartość przed transformacją logistyczną.
predict_proba
implementację z sklearn
interfejsu API: github.com/dmlc/xgboost/blob/master/python-package/xgboost/…