我正在做一个项目,我需要能够根据导入到R中的数据集制定一些决策树。使用rpart包,我希望能够创建一对决策树,一个决策树使用基尼分割标准,另一个使用熵分割标准。如果可能,我还希望能够调整最大树的深度。有人知道怎么做这个吗?我将不胜感激任何帮助!

最佳答案

根据R手册here,可以将rpart()设置为使用基尼或使用以下参数拆分信息(即熵):

parms = list(split = "gini"))

或者
parms = list(split = "information"))

... 分别。

您还可以添加rpart.control的参数(请参阅here),包括maxdepth,其默认值为30

关于r - 用rpart在R中绘制决策树,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/15044526/

10-12 19:52