用r在r上绘制决策树

Question

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

Answer 1

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

parms = list(split = "gini")) 

或

parms = list(split = "information")) 

...。

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