从H2O中的MOJO模型获取模型指标

时间:2019-08-30 20:38:55

标签: scala h2o sparkling-water

我有一个要用于模型指标(rmse,roc等)的MOJO模型,我知道所有模型指标均可用于二元模型,但我想从MOJO模型中获取这些指标。

输入-Mojo模型和训练数据集 输出-提取模型指标

尝试使用以下命令从二进制模型中提取指标: ModelMetricsSupport.modelMetrics [ModelMetricsSupervised](binaryModel,有效)

但是无法对MOJO模型执行相同操作。

1 个答案:

答案 0 :(得分:0)

我能够从Python(H2O)的MOJO模型中获取模型指标,可以通过以下两种方式完成: 一种。 mojo模型的导出zip具有包含所有度量标准信息的模型json。 路径:/experimental/modelDetails.json

b。第二种方法是导入和调用model_performance()

导入h2o

h2o.init()

model = h2o.import_mojo(“ / Users / sonamsharma / delete / GLM_model_python_1567539104058_1.zip”)

model.model_performance()

这可以按原样使用,也可以提取json。

对于JAVA,我尝试参考以下问题修复和代码,但无法做到这一点。 https://github.com/h2oai/h2o-3/pull/3642