r - R中的随机森林 - 许多类

Question

我想用 R randomForest 进行多标签分类。我有十个班A..J，

我找到了如何预测单个类的示例，例如：

r = randomForest(J ~., data=train, importance=TRUE, do.trace=100)

但我想预测更多类，例如 H、I、J。（即说只有 A..G 被赋予属性）。我该怎么做？

我有一个保留 A..G 和仅一个预测类（H/I/J）并运行 randomForest 3 次的想法，但也许有更好的方法？一口气做完？

提前谢谢了。

Answer 1

假设所有属性 H、I 和 J 都是二元的。然后你可以预测一个具有 2^3 个可能值的新属性 K，然后将结果解码回 3 个属性：

• 1 -> 0,0,0
• 2 -> 0,0,1
• 3 -> 0,1,0
• 4 -> 0,1,1
• 5 -> 1,0,0
• 6 -> 1,0,1
• 7 -> 1,1,0
• 8 -> 1,1,1