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
才能提前停止。
方法