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