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