从 H2O 模型对象获取模型详细信息

Question

我有一个相当简单的问题，但无法在任何地方找到记录在案的解决方案。

我目前正在使用 H2O 模型构建管道，并且作为过程的一部分，我需要将有关每个训练模型的一些基本信息写入表格。

假设我有类似的东西：

model = H2ODeepLearningEstimator(...)
model.train(...)

这样做之后，我想从model对象中提取模型的类型。即，我正在寻找类似的东西：

model.getType()

然后返回一个字符串"H2ODeepLearningEstimator"或等效的"deeplearning"H2O 似乎在内部用作模型类型标识符。我还想获得其他详细信息，例如它是回归模型还是分类模型。我没有看到公开此信息的参数。

model.save_model_details例如，如果我跑步，我会得到：

H2ODeepLearningEstimator :  Deep Learning
Model Key:  Grid_DeepLearning_py_4_sid_a02a_model_python_1502450758585_2_model_0


ModelMetricsRegression: deeplearning
** Reported on train data. **

MSE: 19.5334650304
RMSE: 4.4196679774
MAE: 1.44489752843
RMSLE: NaN
Mean Residual Deviance: 19.5334650304

ModelMetricsRegression: deeplearning
** Reported on validation data. **
...
...

大概是model.save_model_details根据各个参数建立此摘要。我想通过直接访问这些（以及类似）参数model对象（性能指标这是可能的通过model.mse()，model.mae()等等）

Answer 1

h2o.algo 为您提供模型类型。至于回归或分类，我不知道，但它是他们的某个地方。查看流程，因为更容易看到参数名称或模型。并滚动浏览，直到看到看起来可能有它的东西。