使用rpart在R中绘制决策树

Question

5

我正在处理一个项目，需要根据我导入到R中的数据集创建一些决策树。使用rpart软件包，我想创建一对决策树，其中一个使用基尼系数拆分标准，另一个使用熵拆分标准。如果可能的话，我还想调整最大树深度。有人知道如何做到这一点吗？非常感谢任何帮助！

- Hoser

如果你附上一个可复制的数据集，其他人可以把它作为参考，那就太好了。请参考 https://dev59.com/eG025IYBdhLWcg3whGSx 获取指南。 - Jochem

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Simon · Accepted Answer

根据 R 手册 here，rpart() 可以使用参数来设置使用基尼系数或信息熵进行分割：

parms = list(split = "gini"))

或者

parms = list(split = "information"))

...分别。

您还可以为rpart.control添加参数（请参见此处），包括maxdepth，默认值为30。