lightgbm.plot_split_value_histogram

lightgbm.plot_split_value_histogram(booster, feature, bins=None, ax=None, width_coef=0.8, xlim=None, ylim=None, title='Split value histogram for feature with @index/name@ @feature@', xlabel='Feature split value', ylabel='Count', figsize=None, dpi=None, grid=True, **kwargs)[source]

绘制模型中指定特征的分割值直方图。

参数:
  • booster (BoosterLGBMModel) – 需要绘制特征分割值直方图的 Booster 或 LGBMModel 实例。

  • feature (intstr) – 绘制直方图所对应的特征名称或索引。如果为 int,则解释为索引。如果为 str,则解释为名称。

  • bins (int, strNone, 可选 (默认值=None)) – 最大 bin 数量。如果为 None,则 bin 数量等于唯一的分割值数量。如果为 str,则应是 numpy.histogram() 函数支持的值列表中的一个。

  • ax (matplotlib.axes.AxesNone, 可选 (默认值=None)) – 目标坐标轴实例。如果为 None,将创建新的图和坐标轴。

  • width_coef (float, 可选 (默认值=0.8)) – 直方图条形宽度的系数。

  • xlim (包含 2 个元素的 tupleNone, 可选 (默认值=None)) – 传递给 ax.xlim() 的 tuple。

  • ylim (包含 2 个元素的 tupleNone, 可选 (默认值=None)) – 传递给 ax.ylim() 的 tuple。

  • title (strNone, 可选 (默认值="Split value histogram for feature with @index/name@ @feature@")) – 坐标轴标题。如果为 None,则禁用标题。可以使用 @feature@ 占位符,它将被替换为 feature 参数的值。可以使用 @index/name@ 占位符,当 feature 参数类型为 int 时,它将被替换为 index;当 feature 参数类型为 str 时,它将被替换为 name

  • xlabel (strNone, 可选 (默认值="Feature split value")) – X 轴标题标签。如果为 None,则禁用标题。

  • ylabel (strNone, 可选 (默认值="Count")) – Y 轴标题标签。如果为 None,则禁用标题。

  • figsize (包含 2 个元素的 tupleNone, 可选 (默认值=None)) – 图的大小。

  • dpi (intNone, 可选 (默认值=None)) – 图的分辨率。

  • grid (bool, 可选 (默认值=True)) – 是否为坐标轴添加网格。

  • **kwargs – 传递给 ax.bar() 的其他参数。

返回值:

ax – 包含指定模型的特征分割值直方图的图。

返回类型:

matplotlib.axes.Axes