
    &`i                     (   d Z ddlZddlZddlmZ d Zd Zedk    rddlZ ej                    Z	e	
                    ddd	
           e	                                \  ZZ ej        d            ej        e ej        ddej        rdnd          ej        rdnd ej        dd           ej        dd           ej        ddg          d          Ze                                Z ede                                j                   dS dS )zGThis example demonstrates basic Ray Tune random search and grid search.    N)tunec                 R    t          j        d           d|| z  dz  z   dz  |dz  z   S )Ng?d   )timesleep)stepwidthheights      x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/examples/tune_basic_example.pyevaluation_fnr      s1    JsOOO%$,$$"-<<    c                     | d         | d         }}t          | d                   D ]*}t          |||          }t          j        ||d           +d S )Nr
   r   steps)
iterations	mean_loss)ranger   r   report)configr
   r   r	   intermediate_scores        r   easy_objectiver      sk    7OVH%56EfWo&& K K*4??46HIIJJJJ	K Kr   __main__z--smoke-test
store_truezFinish quickly for testing)actionhelpF)configure_loggingr   min   2   )metricmodenum_samplesr      irelutanh)r   r
   r   
activation)tune_configparam_spacez!Best hyperparameters found were: )__doc__r   rayr   r   r   __name__argparseArgumentParserparseradd_argumentparse_known_argsargs_initTuner
TuneConfig
smoke_testuniformgrid_searchtunerfitresultsprintget_best_resultr    r   r   <module>r?      s   M M  



      = = =
K K K zOOO$X$&&F
|2N     %%''GD!CHu%%%% DJ#DO!_4"
 
 
 /2QQs!T\!R(("dl4--*$*FF+;<<	
 
  E iikkG	E
-w/F/F/H/H/OPPPPPC r   