我正在使用 H2O 苏打水来构建 GBM 模型。我知道我们可以使用下面的代码查看 N 文件夹交叉验证结果:
gbm_model.model_performance(xval = True)
但是有没有办法将每个文件夹的模型性能保存到数据框中?例如,将每个文件夹的 AUC 保存到数据框中。
我正在使用 H2O 苏打水来构建 GBM 模型。我知道我们可以使用下面的代码查看 N 文件夹交叉验证结果:
gbm_model.model_performance(xval = True)
但是有没有办法将每个文件夹的模型性能保存到数据框中?例如,将每个文件夹的 AUC 保存到数据框中。
创建模型时,您可以通过调用模型对象来设置keep_cross_validation_models=True
然后获取模型。cross_validation_models()
然后您可以照常获取每个模型的性能。请参阅文档。