我使用包树生成了一个决策树。 我正在使用 maptree 绘图。我的代码如下所示:
library(tree)
library(maptree)
treeTree = tree(fmla,data = df, control = tree.control(n,mincut = 50))
prune.tree(treeTree, eps = 10^(-3) )
pdf( paste(fName, varName,"tree", "tree.pdf", sep = "_") )
missc = deviance(treeTree, detail=TRUE)
text(treeTree, splits = TRUE, label = "yval", all = FALSE,
pretty = NULL, digits = getOption("digits") - 3,adj =
par("adj"), xpd = TRUE)
title(main=varName)
dev.off()
不幸的是,这似乎绘制了每个终端节点的中位数。所以我得到一棵看起来像这样的树:
| \
| \
| \
NO NO
| \
| \
| \
NO NO
即。很多分裂,但我们总是在NO。
这是因为对于我的结果变量(代表患病患者的风险),中位风险总是为0.
我想在每个节点上显示平均值,或者甚至更好地显示每个选项的结果部分。
我已阅读maptree手册,但无法弄清楚如何做到这一点。