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都是二元的。然后，你可以预测一个新的属性K的2^3个可能的值，然后将结果解码成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