ray.tune.stopper.ExperimentPlateauStopper#

class ray.tune.stopper.ExperimentPlateauStopper(metric: str, std: float = 0.001, top: int = 10, mode: str = 'min', patience: int = 0)[source]#

Bases: Stopper

Early stop the experiment when a metric plateaued across trials.

Stops the entire experiment when the metric has plateaued for more than the given amount of iterations specified in the patience parameter.

Parameters:

  • metric – The metric to be monitored.

  • std – The minimal standard deviation after which the tuning process has to stop.

  • top – The number of best models to consider.

  • mode – The mode to select the top results. Can either be “min” or “max”.

  • patience – Number of epochs to wait for a change in the top models.

Raises:

  • ValueError – If the mode parameter is not “min” nor “max”.

  • ValueError – If the top parameter is not an integer greater than 1.

  • ValueError – If the standard deviation parameter is not a strictly positive float.

  • ValueError – If the patience parameter is not a strictly positive integer.

Methods

stop_all

Return whether to stop and prevent trials from starting.