ray.tune.stopper.TrialPlateauStopper#

class ray.tune.stopper.TrialPlateauStopper(metric: str, std: float = 0.01, num_results: int = 4, grace_period: int = 4, metric_threshold: float | None = None, mode: str | None = None)[源代码]#

基类:Stopper

当单次试验达到平台期时,提前停止。

当试验的 metric 结果的标准偏差低于阈值 std 时,试验将达到平稳状态并提前停止。

参数:
  • metric – 用于检查收敛的指标。

  • std – 决定试验是否达到平台期的最大指标标准差。默认为 0.01。

  • num_results – 用于标准差计算的结果数量。

  • grace_period – 在试验可以被提前停止之前的最小时间步数

  • metric_threshold (Optional[float]) – 结果必须超过的最小或最大值,之后才能提前停止。

  • mode – 如果传递了 metric_threshold 参数,这必须是 [min, max] 之一。指定我们是优化一个大的指标(max)还是一个小的指标(min)。如果是 max,metric_threshold 必须被超过,如果是 min,值必须低于 metric_threshold 才能提前停止。

方法