首页 > 解决方案 > 将 H2O AutoML 模型转换为 Sklearn 模型

问题描述

我有一个 H2O AutoML 使用 python 生成的 GBM 模型。我想知道我们是否可以将其转换为标准的 sklearn 模型,以便我可以将其融入我的其他 sklearn 模型的生态系统中。打印模型时,我可以看到如下模型属性。

在此处输入图像描述

如果从 H2O 直接转换到 sklearn 不可行,有没有办法可以使用上述属性在 sklearn 中重新创建 GBM?这些术语看起来与标准 sklearn GBM 参数略有不同。

提前致谢。

标签: pythonscikit-learnh2oautomlboosting

解决方案


这会有点棘手,因为包有点不同。Sklearn 基于 Python/Cython/C,H2O 使用 Java。底层算法也可能不同。但是,您可以尝试在两者之间匹配/转换您的超参数,因为它们会相似。

此外,最好有一个与库无关的生态系统,以便您可以互换不同的模型。


推荐阅读