h2o.auc（perf，xval = TRUE） - 這個調用返回什麼？

Question

我的代碼如下

gbm.fit.hex = h2o.gbm(x= xcols , y =1865 , training_frame = tr.hex , distribution = "bernoulli", model_id = "gbm.model" , key = "gbm.model.key" ,     ntrees = gbm.trees , max_depth = gbm.depth , min_rows = gbm.min.rows ,     learn_rate = gbm.learn.rate , nbins = 20 , balance_classes = gbm.balance , nfolds = gbm.folds) 

perf <- h2o.performance(gbm.fit.hex , tr.hex) a = h2o.auc(perf , xval = TRUE)

什麼的AUC調用返回？它是否返回訓練數據集或交叉驗證結果的AUC？

Answer 1

它檢索交叉驗證的AUC。

由於您將nfolds參數設置爲非零，因此h2o.gbm函數除了在完整訓練集上訓練GBM模型外，還執行k倍交叉驗證。在你的命令中，你沒有指定驗證集，所以你可以檢索的AUC值是訓練AUC，h2o.auc(perf, train = TRUE)和交叉驗證的AUC（如上）。

如果要在單獨驗證（或測試）集上評估性能，可以使用validation_frame參數傳遞該幀，並使用h2o.auc(perf, valid = TRUE)檢索驗證AUC。