h2o.auc( perf , xval =TRUE) - 这叫什么 return？

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 调用什么 return？ return 是训练数据集上的 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。

h2o.auc( perf , xval =TRUE) - 这叫什么 return？

h2o.auc( perf , xval =TRUE) - what does this call return?

h2o