我正在做一个项目,我需要能够根据导入到R中的数据集制定一些决策树。使用rpart包,我希望能够创建一对决策树,一个决策树使用基尼分割标准,另一个使用熵分割标准。如果可能,我还希望能够调整最大树的深度。有人知道怎么做这个吗?我将不胜感激任何帮助!
最佳答案
根据R手册here,可以将rpart()
设置为使用基尼或使用以下参数拆分信息(即熵):
parms = list(split = "gini"))
或者
parms = list(split = "information"))
... 分别。
您还可以添加
rpart.control
的参数(请参阅here),包括maxdepth
,其默认值为30
。关于r - 用rpart在R中绘制决策树,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/15044526/