lightgbm.plot_metric

lightgbm.plot_metric(booster, metric=None, dataset_names=None, ax=None, xlim=None, ylim=None, title='Metric during training', xlabel='Iterations', ylabel='@metric@', figsize=None, dpi=None, grid=True)[源码]

绘制训练过程中的一个指标。

参数:

  • booster (dict 或 LGBMModel) – 从 lightgbm.train() 返回的字典或 LGBMModel 实例。

  • metric (str 或 None, 可选 (默认值=None)) – 要绘制的指标名称。仅支持一个指标，因为不同指标具有不同的尺度。如果为 None，则从字典中选择第一个指标（根据哈希码）。

  • dataset_names (str 列表, 或 None, 可选 (默认值=None)) – 用于计算要绘制指标的数据集名称列表。如果为 None，则使用所有数据集。

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

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

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

  • title (str 或 None, 可选 (默认值="训练期间的指标")) – 坐标轴标题。如果为 None，则禁用标题。

  • xlabel (str 或 None, 可选 (默认值="迭代次数")) – X 轴标题标签。如果为 None，则禁用标题。

  • ylabel (str 或 None, 可选 (默认值="@metric@")) – Y 轴标题标签。如果为 ‘auto’，则使用指标名称。如果为 None，则禁用标题。可以使用 @metric@ 占位符，它将被替换为指标名称。

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

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

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

返回值:

ax – 绘制了训练过程中指标历史的图。

返回类型:

matplotlib.axes.Axes