&`i RddlZddlmZddlmZeGddeZdS)N)Stopper) PublicAPIc JeZdZdZ ddededed ed ef d Zd Zd Z dZ 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? minrmetricstdtopmodepatiencec|dvrtdt|tr|dkrtdt|tr|dkrtdt|tr|dkrtd||_||_||_d|_||_||_ g|_ dS)N)r maxz1The 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)selfr r r r rs w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/stopper/experiment_plateau.py__init__z!ExperimentPlateauStopper.__init__#s ~ % %PQQ Q#s## saxx8 (C(( NHqLLLMM M#u%% R   !  c~|j||j|jdkr't |jd|j|_n't |j|j d|_|r|xjdz c_nd|_|S)z8Return a boolean representing if the tuning has to stop.r Nrr) rappendrrsortedr has_plateauedrstop_all)rtrial_idresults r__call__z!ExperimentPlateauStopper.__call__@s t| 4555 :  %d&677 $) DD  %d&677 ED       !    !    !D }}r ct|j|jko!tj|j|jkS)N)lenrrnpr rrs rr$z&ExperimentPlateauStopper.has_plateauedTs5  ! !TY . X26$:J3K3Kty3X r cJ|o|j|jkS)z8Return whether to stop and prevent trials from starting.)r$rrr,s rr%z!ExperimentPlateauStopper.stop_allYs#!!##J(8DN(JJr N)rrr r) __name__ __module__ __qualname____doc__strrrrr(r$r%r rrrs:      :(   KKKKKr r)numpyr+ray.tune.stopper.stopperrray.util.annotationsrrr3r rr7s,,,,,,****** SKSKSKSKSKwSKSK SKSKSKr