lightgbm.plot_importance

lightgbm.plot_importance(booster, ax=None, height=0.2, xlim=None, ylim=None, title='Feature importance', xlabel='Feature importance', ylabel='Features', importance_type='auto', max_num_features=None, ignore_zero=True, figsize=None, dpi=None, grid=True, precision=3, **kwargs)[source]

绘制模型的特征重要性。

参数:
  • booster (BoosterLGBMModel) – 需要绘制特征重要性的 Booster 或 LGBMModel 实例。

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

  • height (float, 可选 (默认值=0.2)) – 条形高度,传递给 ax.barh()

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

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

  • title (strNone, 可选 (默认值="Feature importance")) – 坐标轴标题。如果为 None,则禁用标题。

  • xlabel (strNone, 可选 (默认值="Feature importance")) – X 轴标题标签。如果为 None,则禁用标题。可以使用占位符 @importance_type@,它将被替换为 importance_type 参数的值。

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

  • importance_type (str, 可选 (默认值="auto")) – 重要性计算方式。如果为 "auto",则当 booster 参数为 LGBMModel 时,使用 booster.importance_type 属性;否则使用 "split"。如果为 "split",结果包含特征在模型中被使用的次数。如果为 "gain",结果包含使用该特征进行分割的总增益。

  • max_num_features (intNone, 可选 (默认值=None)) – 在图上显示的前 Top 个特征的最大数量。如果为 None 或 <1,则显示所有特征。

  • ignore_zero (bool, 可选 (默认值=True)) – 是否忽略重要性为零的特征。

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

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

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

  • precision (intNone, 可选 (默认值=3)) – 用于将浮点值的显示限制到特定精度。

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

返回值:

ax – 包含模型特征重要性的图。

返回类型:

matplotlib.axes.Axes