dtreeviz 中的子树可视化答案

【问题标题】：Subtrees visualization in dtreevizdtreeviz 中的子树可视化
【发布时间】：2021-09-18 23:42:20
【问题描述】：

我是 dtreeviz 的新手。我正在努力处理一个非常难以可视化的非常深的决策树（过度拟合对我的任务来说不是问题）。我想知道是否有办法只可视化这三个节点中的一些节点（例如，前 5 个节点）。

谢谢！

from dtreeviz.models.xgb_decision_tree import ShadowXGBDTree
from dtreeviz import trees 

xgb_shadow = ShadowXGBDTree(xgb_model_reg, 0, d.loc[:, d.columns != output_quantitativi[0]], 
                             d[output_quantitativi[0]], d.loc[:, d.columns != output_quantitativi[0]].columns,output_quantitativi[0])

trees.dtreeviz(xgb_shadow)

【问题讨论】：

标签： machine-learning data-visualization decision-tree xgboost dtreeviz

【解决方案1】：

对于 dtreeviz 方法，它只是添加了参数 depth_range_to_display，它允许您指定要显示的树级别范围。

对于 viz_leaf_samples()，如果树包含大量叶节点，您可以使用 min_samples 和 max_samples 值。

【讨论】：