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)[源代码]

绘制模型的特征重要性。

参数:

  • booster (Booster or LGBMModel) – 应绘制其特征重要性的 Booster 或 LGBMModel 实例。

  • ax (matplotlib.axes.Axes or None, optional (default=None)) – 目标轴实例。如果为 None，将创建新的图形和轴。

  • height (float, optional (default=0.2)) – 条形高度，传递给 ax.barh()。

  • xlim (tuple of 2 elements or None, optional (default=None)) – 传递给 ax.xlim() 的元组。

  • ylim (tuple of 2 elements or None, optional (default=None)) – 传递给 ax.ylim() 的元组。

  • title (str or None, optional (default="Feature importance")) – 坐标轴标题。如果为 None，则标题将被禁用。

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

  • ylabel (str or None, optional (default="Features")) – Y轴标题标签。如果为None，则标题被禁用。

  • importance_type (str, optional (default="auto")) – 重要性是如何计算的。如果为“auto”，如果 booster 参数是 LGBMModel，则使用 booster.importance_type 属性；否则为“split”。如果为“split”，结果包含特征在模型中使用的次数。如果为“gain”，结果包含使用该特征的分割的总增益。

  • max_num_features (int or None, optional (default=None)) – 图表上显示的最大顶级特征数量。如果为 None 或 <1，将显示所有特征。

  • ignore_zero (bool, optional (default=True)) – 是否忽略重要性为零的特征。

  • figsize (tuple of 2 elements or None, optional (default=None)) – 图表尺寸。

  • dpi (int or None, optional (default=None)) – 图像的分辨率。

  • grid (bool, optional (default=True)) – 是否为轴添加网格。

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

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

返回:

ax – 带有模型特征重要性的图表。

返回类型:

matplotlib.axes.Axes