r - 如何在 h2o 随机森林中指定分类

Question

我正在互联网上寻找如何在 h2o.randomForest 中指定分类。无论我能找到什么，我们都可以将参数“分类”设置为 TRUE。但根据 h2o 包文档，它不再存在。

当我在目标变量是二进制 1-0 变量的数据上运行这个 h2o.randomForest 时，它假设回归。

我不知道如何告诉它我想做分类。

任何帮助将不胜感激。

谢谢

score 2 · Accepted Answer

如果您使用 R，则需要使用 as.factor 来转换二进制 1-0 变量，如果您使用 Flow，则需要将二进制 1-0 变量转换为枚举。完成此操作后，H2O 的随机森林会自动将此识别为分类问题。

谢谢，

阿维尼

1 回答 1