我在R中使用h2o
将gbm模型拟合到我的数据中。我有几个分类变量,我正在使用h2o.gbm
函数中的选项categorical_encoding
对这些变量进行编码。
在处理测试集中的预测时,如何处理分类变量?函数predict
似乎没有自动编码分类变量的选项。
H2O 的好处是你不必担心手动分类编码。h2o.predict()
将应用在训练期间指定的相同分类编码方法。有关如何在训练期间设置分类编码类型的详细信息,请参阅文档。