o $i @s8ddlZddlmZddlmZeGdddeZdS)N)Stopper) PublicAPIc @sPeZdZdZ    ddededed ed ef d d Zd dZddZ ddZ dS)ExperimentPlateauStopperaEarly 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. Args: 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. MbP? minrmetricstdtopmodepatiencecCs|dvrtdt|tr|dkrtdt|tr|dkr"tdt|tr+|dkr/td||_||_||_d|_||_||_ g|_ dS)N)rmaxz1The mode parameter can only be either min or max.zDTop results to consider must be a positive integer greater than one.rz-Patience must be a strictly positive integer.z@The standard deviation must be a strictly positive float number.) ValueError isinstanceintfloat_mode_metric _patience _iterations_std_top _top_values)selfrr r r r r`/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/tune/stopper/experiment_plateau.py__init__#s& z!ExperimentPlateauStopper.__init__cCsx|j||j|jdkrt|jd|j|_n t|j|j d|_|r5|jd7_|Sd|_|S)z8Return a boolean representing if the tuning has to stop.rNrr) rappendrrsortedr has_plateauedrstop_all)rtrial_idresultrrr__call__@s z!ExperimentPlateauStopper.__call__cCs"t|j|jkot|j|jkS)N)lenrrnpr rrrrrr Ts z&ExperimentPlateauStopper.has_plateauedcCs|o |j|jkS)z8Return whether to stop and prevent trials from starting.)r rrr'rrrr!Ysz!ExperimentPlateauStopper.stop_allN)rrrr) __name__ __module__ __qualname____doc__strrrrr$r r!rrrrrs(  r)numpyr&ray.tune.stopper.stopperrray.util.annotationsrrrrrrs