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 or LGBMModel) – 从 lightgbm.train() 返回的字典或 LGBMModel 实例。

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

  • dataset_names (list of str, or None, optional (default=None)) – 用于计算指标并绘图的数据集名称列表。如果为 None，则使用所有数据集。

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

  • 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="Metric during training")) – 坐标轴标题。如果为 None，则标题将被禁用。

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

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

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

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

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

返回:

ax – 训练过程中指标历史记录的图表。

返回类型:

matplotlib.axes.Axes