【发布时间】:2021-11-20 21:14:20
【问题描述】:
我正在一个项目中工作,该项目需要在 R 中使用 Rpart 包来构建决策树。
但是,我的直接下属经理对作为树中根节点的特定变量感兴趣,因为他认为该特定变量对我们的业务很重要。
我是否知道有人知道如何强制树以该特定变量开始?例如,该变量名是 X。
数据集描述:目标变量 Y. 与其他 30 个自变量。
代码:
tree <- rpart(Y~., method="class", data=train, parms=list(split="information"), control=rpart.control(cp=.0002,minsplit = 5,minbucket = 5,maxdepth = 10))
【问题讨论】:
-
你需要什么?拆分规则是否足够?你需要一张树的照片吗?您需要能够交叉验证还是训练错误足够好?
-
您好,每个参数都足够好。我能够构建树并能够生成情节。但是,我想强制它以 X 开头的树的第一个根节点。请问您是否知道如何在代码中执行此操作?谢谢
标签: r decision-tree rpart