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值。