R 中的預測給出概率或分數？

Question

我是 R 的新手並嘗試構建 xgboost model，我能夠在 pred 變量中獲得預測

X_train <- train_list$X
X_test <- test_list$X

X_Train <- as(X_train, "dgCMatrix")



xgb <- xgboost(data = X_train, 
               label = y_train, 
               eta = 0.9,
               max_depth = 3, 
               nround=10, 
               subsample = 0.3,      
               colsample_bytree = 0.1,  
               seed = 1,
               eval_metric = "auc",
               objective = "binary:logistic",
               #num_class = 2,
               booster = 'gbtree',  
               max_delta_step = 1,    
               scale_pos_weight=2,   
               gamma=5 ,                  
               min_child_weight = 2    
)


pred <- predict(xgb, X_test)

我想知道 pred 是概率還是分數，我覺得預測是概率但不確定。

如果有人可以對此表示懷疑，如果不是概率，那么它是什么？

Answer 1

在二元/2 類問題中，如果其中一個類（正）= 1，而另一個 class（負）= 0，則您使用經過訓練的 model（通過predict() ）計算的預測將介於 0 之間和 1. 預測值為 1（正類）或 0（負類）的概率是與 0.5 的絕對差，您通常轉換預測，使 <0.5 變為“0”（負），>0.5 變為“ 1'（陽性）。 這一切都在文檔中詳細解釋： https://xgboost.readthedocs.io/en/latest/R-package/xgboostPresentation.html#perform-the-prediction