lightgbm.create_树_图

lightgbm.create_tree_digraph(booster, tree_index=0, show_info=None, precision=3, orientation='horizontal', example_case=None, max_category_values=10, **kwargs)

创建指定树的有向图表示。

图中的每个节点代表树中的一个节点。

非叶节点有类似 Column_10 <= 875.9 的标签，这意味着“此节点根据名为“Column_10”的特征进行分割，阈值为875.9”。

叶节点有类似 leaf 2: 0.422 的标签，这意味着“此节点是叶节点，落入此节点的记录的预测值为 0.422”。数字（2）是一个内部唯一标识符，没有任何特殊含义。

备注

更多信息请访问 https://graphviz.readthedocs.io/en/stable/api.html#digraph。

参数:

• booster (Booster or LGBMModel) – 要转换的 Booster 或 LGBMModel 实例。

• tree_index (int, optional (default=0)) – 要转换的目标树的索引。

• show_info (list of str, or None, optional (default=None)) –

  节点中应显示哪些信息。

  • 'split_gain' : 将此分割添加到模型中获得的增益

  • 'internal_value' : 如果该节点是叶节点，则生成的原始预测值

  • 'internal_count' : 训练数据中落入此非叶节点的记录数量

  • 'internal_weight' : 所有落入此非叶节点的节点的总权重

  • 'leaf_count' : 训练数据中落入此叶节点的记录数

  • 'leaf_weight' : 落入此叶节点的所有观测值的总权重（Hessian 的和）

  • 'data_percentage' : 落入此节点的训练数据百分比

• precision (int or None, optional (default=3)) – 用于将浮点数值的显示限制在一定的精度范围内。

• orientation (str, optional (default='horizontal')) – 树的方向。可以是 ‘horizontal’ 或 ‘vertical’。

• example_case (numpy 2-D array, pandas DataFrame or None, optional (default=None)) – 具有与训练数据相同结构的单行。如果不是 None，图表将突出显示该样本在树中的路径。 .. versionadded:: 4.0.0

• max_category_values (int, optional (default=10)) – 在树节点中显示的类别值的最大数量，如果阈值数量大于此值，阈值将被折叠并在标签工具提示中显示。 .. warning:

• **kwargs – 传递给 Digraph 构造函数的其他参数。查看 https://graphviz.readthedocs.io/en/stable/api.html#digraph 获取支持参数的完整列表。

返回:

graph – 指定树的有向图表示。

返回类型:

graphviz.Digraph