r - h2o 中的分类编码 - 测试集

Question

我h2o在 R 中使用 gbm 模型来拟合我的数据。我有几个分类变量，我正在使用函数中的选项categorical_encoding对h2o.gbm这些变量进行编码。

在处理测试集中的预测时，我如何处理分类变量？该函数predict似乎没有自动编码分类变量的选项。

Answer 1

H2O 的好处是您不必担心手动分类编码。h2o.predict()将应用在训练期间指定的相同分类编码方法。有关如何在训练期间设置分类编码类型的更多信息，请参阅文档。