当我的因变量中有4个类别时,想要构建决策树

San*_*una 6 cart decision-tree rstudio

我想构建一个决策树,但我的因变量中有4个类别(1,2,3,4).我该如何建造?我熟悉rpart包,它可以用于二进制因变量.我想如果我们有更多类别,那么我们需要构建CHAID模型.请建议.

data<-rpart(cluster, indvariable, data=segmentation, control=rpart.control(cp=0.01)
Run Code Online (Sandbox Code Playgroud)

Cluster是我的因变量,有4个类别.

提前致谢.

Jos*_*osh 0

我不认为做多类模型与二元分类有什么不同,你说你已经知道了。

你尝试过这样的语法吗?

treemodel = rpart(class ~ x1 + x2 + x3 + x4 + x5, data = df)
Run Code Online (Sandbox Code Playgroud)