首页 > 解决方案 > R中实现的H2O AutoML(3.24.0.5)中多类分类的“metalearner”问题

问题描述

我正在尝试 H2O AutoML 进行多类分类。一切似乎都运行良好,除非我试图从 Metalearner 中提取 StackedEnsemble 的变量重要性。

这是我从代码中得到的:

# Get the "BestOfFamily" Stacked Ensemble model
se.best <- h2o.getModel(grep("StackedEnsemble_BestOfFamily_AutoML", model_ids, value = TRUE)[1])
#se.best
metalearner <- h2o.getModel(se.best@model$metalearner$name)
#as.data.frame(h2o.varimp(metalearner))
h2o.varimp_plot(metalearner, num_of_features = 20)

metalearner 对“StackedEnsemble_BestOfFamily_AutoML”的可变重要性:

在此处输入图像描述

这里的问题不是显示 H2O AutoML 选择的 20 个模型名称(来自“max_models = 20”),上图显示了多类分类的类的名称。

对我来说,这看起来像一个错误,因为当我在 R 中尝试使用 H2O AutoML 进行回归时,我没有遇到这个问题。

标签: rh2oautoml

解决方案


推荐阅读