visTree：决策树子群的可视化
应用统计
2021-03-09 v1 其他统计学
摘要
决策树是灵活的预测模型，用于量化结局-协变量关系并刻画相关人群子群。然而，拟合决策树的标准图形表示强调单个分割点，因而对于可视化已定义子群并非最优。本文提出一种决策树的新颖视觉表示，将主要焦点转向刻画子群——既包括其定义协变量，也包括其结局分布。我们在 \texttt{visTree} 包中实现了该方法，其构建于 \texttt{partykit} 包所提供的工具集与基础设施之上，并可应用于多种决策树的可视化。利用来自 Box Lunch 研究 [French et al., 2014]（一项评估不同午餐分量暴露对在职成人能量摄入与体重影响的随机试验）的数据，演示了各独立函数。
引用
@article{arxiv.2103.04139,
title = {visTree: Visualization of Subgroups for a Decision Tree},
author = {Ashwini Venkatasubramaniam and Julian Wolfson},
journal= {arXiv preprint arXiv:2103.04139},
year = {2021}
}