R中的Maptree决策树图:显示每个节点的分布

时间:2017-06-27 12:28:00

标签: r tree classification mean median

我使用包生成了一个决策树。 我正在使用 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手册,但无法弄清楚如何做到这一点。

0 个答案:

没有答案