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 (Booster 或 LGBMModel) – 需要绘制特征分割值直方图的 Booster 或 LGBMModel 实例。
feature (int 或 str) – 绘制直方图所对应的特征名称或索引。如果为 int,则解释为索引。如果为 str,则解释为名称。
bins (int, str 或 None, 可选 (默认值=None)) – 最大 bin 数量。如果为 None,则 bin 数量等于唯一的分割值数量。如果为 str,则应是
numpy.histogram()
函数支持的值列表中的一个。ax (matplotlib.axes.Axes 或 None, 可选 (默认值=None)) – 目标坐标轴实例。如果为 None,将创建新的图和坐标轴。
width_coef (float, 可选 (默认值=0.8)) – 直方图条形宽度的系数。
xlim (包含 2 个元素的 tuple 或 None, 可选 (默认值=None)) – 传递给
ax.xlim()
的 tuple。ylim (包含 2 个元素的 tuple 或 None, 可选 (默认值=None)) – 传递给
ax.ylim()
的 tuple。title (str 或 None, 可选 (默认值="Split value histogram for feature with @index/name@ @feature@")) – 坐标轴标题。如果为 None,则禁用标题。可以使用 @feature@ 占位符,它将被替换为
feature
参数的值。可以使用 @index/name@ 占位符,当feature
参数类型为int
时,它将被替换为index
;当feature
参数类型为str
时,它将被替换为name
。xlabel (str 或 None, 可选 (默认值="Feature split value")) – X 轴标题标签。如果为 None,则禁用标题。
ylabel (str 或 None, 可选 (默认值="Count")) – Y 轴标题标签。如果为 None,则禁用标题。
figsize (包含 2 个元素的 tuple 或 None, 可选 (默认值=None)) – 图的大小。
dpi (int 或 None, 可选 (默认值=None)) – 图的分辨率。
grid (bool, 可选 (默认值=True)) – 是否为坐标轴添加网格。
**kwargs – 传递给
ax.bar()
的其他参数。
- 返回值:
ax – 包含指定模型的特征分割值直方图的图。
- 返回类型:
matplotlib.axes.Axes