
    &`i
                         d Z ddlZddlZddlmZ ddlmZ ddlm	Z	 d Z
d Zedk    rTddlZ ej                    Ze                    d	d
d           e                                \  ZZ e	dgdg          Zej                            ed          Z e            Z ej        e ej        dddi           ej        ddeeej        rdnd          d ej        dd           ej        dd           ej        dd           ej        dd           ej        dd           ej        dd          d          Ze                                Z e de!                                j"                   dS dS ) zThis example demonstrates the usage of AxSearch with Ray Tune.

It also checks that it is usable with a separate scheduler.

Requires the Ax library to be installed (`pip install ax-platform`).
    N)tune)AsyncHyperBandScheduler)AxSearchc                    t          j        g d          }t          j        g dg dg dg dg          }dt          j        g dg dg d	g d
g          z  }d}t          |          D ]X\  }}d}t          d          D ]&}||||f         | |         |||f         z
  dz  z  z  }'||t          j        |           z  z  }Y|S )N)      ?g333333?g      @g	@)
               @333333?   )皙?r   r
   皙?r      )r	   r   r   r   r
   r   )r
   r   r   r   r   r   g-C6?)i   i  i  |   i[   i  )i	  i'  is   i  i  i'  )i,	  i  i  iC  i  i  )i  i|"  i"  io  iC  i}          r         )nparray	enumeraterangeexp)	xalphaAPyjalpha_jtks	            p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/examples/ax_example.py	hartmann6r$      s   H)))**E
$$$&&&$$$&&&		
	 	A 	RX///000000///		
  	A 	A&& " "
7q 	3 	3A1a4QqTAadG^122AA	Wrvqbzz!!H    c           	      R    t           d                   D ]}t          j         fdt          d          D                       }t          j        |t          |          t          j        |dz                                            d           t          j	        d           d S )N
iterationsc                 f    g | ]-}                     d                     |dz                       .S )zx{}   )getformat).0iconfigs     r#   
<listcomp>z"easy_objective.<locals>.<listcomp>.   s5    HHH!fjja!e!4!455HHHr%   r   r   )timesteps_totalr$   l2normg{Gz?)
r   r   r   r   reportr$   sqrtsumtimesleep)r.   r-   r   s   `  r#   easy_objectiver7   ,   s    6,'(( 	 	HHHHHuQxxHHHII#$&q\\'1a4**,,// 	
 	
 	
 	
4	 	r%   __main__z--smoke-test
store_truezFinish quickly for testing)actionhelpzx1 + x2 <= 2.0zl2norm <= 1.25)parameter_constraintsoutcome_constraints   )max_concurrentaxr0   d   )namestopr$   minr   2   )metricmode
search_alg	schedulernum_samplesr   r   )r'   x1x2x3x4x5x6)
run_configtune_configparam_spacez!Best hyperparameters found were: )#__doc__r5   numpyr   rayr   ray.tune.schedulersr   ray.tune.search.axr   r$   r7   __name__argparseArgumentParserparseradd_argumentparse_known_argsargs_algosearchConcurrencyLimiterrI   Tuner	RunConfig
TuneConfig
smoke_testuniformtunerfitresultsprintget_best_resultr.    r%   r#   <module>ro      sK               7 7 7 7 7 7 ' ' ' ' ' '  6
 
 
 zOOO$X$&&F
|2N     %%''GD!8/0-.  D
 ;))$q)AAD''))IDJ!4>#S)
 
 
 $DO"o52
 
 
 $,sC(($,sC(($,sC(($,sC(($,sC(($,sC((
 
  E. iikkG	E
-w/F/F/H/H/OPPPPPQ r%   