r - 用 rpart 在 R 中绘制决策树

Question

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

Answer 1

根据此处的R手册，rpart()可以使用参数设置使用gini或信息（即熵）拆分：

parms = list(split = "gini"))

或者

parms = list(split = "information"))

... 分别。

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