使用rpart绘制R中的决策树

Question

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

Answer 1

根据这里的R手册,rpart()可以设置使用gini或信息(即熵)拆分使用参数:

parms = list(split = "gini"))

要么

parms = list(split = "information"))

... 分别.

您还可以添加rpart.control(参见此处)参数,包括maxdepth默认值30.