scala - 如何使用 Scala 从苏打水中的 H2O AutoML 中排除算法

Question

我必须从 AutoMl 模型中排除一些算法。

我正在尝试排除算法，但它失败了。

buildSpecHopper_1.build_models.exclude_algos = Array(Algo.DeepLearning,Algo.GLM)

但它会抛出 Class cast 异常：

java.lang.ClassCastException: [Lai.h2o.automl.AutoML$algo; cannot be cast to [Lai.h2o.automl.Algo;

Answer 1

在 Sparkling Water 中，使用 AutoML 的正确方法是通过我们的 Spark 包装器： https ://github.com/h2oai/sparkling-water/blob/078f38ae5c863f7203cbdc9c35110f23c557d756/examples/pipelines/hamOrSpamMultiAlgo.script.scala#L97

该包装器具有指定包含和排除算法的选项。

我可以看到您正在直接使用 Java API，并且可能遇到了错误。我们将对此进行研究，但我建议使用更高级别的 API。