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