dtreeviz 中的子树可视化

Question

我是 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)

Answer 1

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

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