r - 不是一棵合法的树

Question

我的数据看起来像这样

> head(main)
  X MYDEPV Price Income Age price20 price30
1 1      1    10     33  37       0       0
2 2      0    20     21  55       1       0

我正在尝试修剪 R 中的决策树。我收到错误“不是合法树”。这个错误是什么意思？什么是合法树？我该如何解决？

> tree <- rpart(MYDEPV ~ Price + Income + Age, main, method="class", parms = list(split = "gini"))
> tree = prune.tree(tree)
Error in prune.tree(tree) : not legitimate tree

Answer 1

我想你想用

prune.rpart()

而不是 prune.tree()

错误“不是合法树”源于 prune.tree() 不期望您在上一行中创建的类“rpart”的对象。

prune.tree 需要“树”类的对象，并且如评论中所述来自树包，而不是 rpart 包。