使用rpart绘制R中的决策树

Question

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

Answer 1

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

parms = list(split = "gini"))

或

parms = list(split = "information"))

......分别。

您还可以添加rpart.control的参数（请参阅here），其中包含maxdepth，其默认值为30。