r语言 - 测试集中的分类编码



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

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

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

最新更新