在partykit包打印(i. ctree)提供了在终端节点(分类树)的结果的概率。然而,我想知道的结果在内部节点的概率以及。
当我为ctree图创建条形图(i.ctree,inner_panel = node_barplot)时,我可以估计内部节点的概率,但我想要的是内部节点的确切概率,例如,在下图中,我想知道节点2和5的结果概率。
有什么想法吗?
以https://cran.r-project.org/web/packages/partykit/vignettes/ctree.pdf为例:
ctree条形图:
1条答案
按热度按时间0yycz8jy1#
有多种方法可以提取属于某个节点的全部数据并计算您感兴趣的任何数量。对于分类树的分布,一种方法是强制到
simpleparty
类,该类将distribution
存储在每个节点的info
槽中。使用您提到的小插图中的示例,首先可以拟合完整的
constparty
树:然后强制为
simpleparty
:然后,您可以从每个节点提取分布列表,将其绑定到表中,并计算比例:
您还可以重新使用
print.simpleparty
中使用的函数,调整面板函数以进行打印: