lightgbm.early_stopping

lightgbm.early_stopping(stopping_rounds, first_metric_only=False, verbose=True, min_delta=0.0)[源代码]

创建一个启用早期停止的回调。

激活提前停止。模型将训练直到验证分数至少提高 min_delta。验证分数需要在每 stopping_rounds 轮中至少提高一次以继续训练。至少需要一个验证数据和一个指标。如果有多个，将检查所有这些。但无论如何都会忽略训练数据。要仅检查第一个指标，请将 first_metric_only 设置为 True。具有最佳性能的迭代索引将保存在模型的 best_iteration 属性中。

参数:

  • stopping_rounds (int) – 在没有趋势发生的情况下可能的回合数。

  • first_metric_only (bool, optional (default=False)) – 是否仅使用第一个指标进行提前停止。

  • verbose (bool, optional (default=True)) – 是否记录带有提前停止信息的消息。默认情况下，使用标准输出资源。使用 register_logger() 函数注册自定义记录器。

  • min_delta (float or list of float, optional (default=0.0)) – 最小分数提升以继续训练。如果是浮点数，则该单一值用于所有指标。如果是列表，其长度应与指标总数匹配。 .. versionadded:: 4.0.0

返回:

回调 – 激活早期停止的回调。

返回类型:

_EarlyStoppingCallback