o ci @s<dZddlZddlZddlmZddlmZddlm Z ddZ dd Z e d krddl Z e Zejd d d de\ZZe dgdgdZejjeddZeZeje ejdddidejddeeejrgdndddeddeddeddeddeddeddddZeZe d e!j"dSdS)!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 Cstgd}tgdgdgdgdg}dtgdgdgd gd g}d }t|D]+\}}d }td D]}||||f|||||fd7}q;||t| 8}q1|S)N)?g333333?g@g @)  @333333?)皙?rr皙?r )rr r rrr )rr r rr rg-C6?)i ii|i[ i)i i'is iii')i, ii iC i i)ii|"i"ioiCi}r)nparray enumeraterangeexp) xalphaAPyjalpha_jtkr!P/home/ubuntu/.local/lib/python3.10/site-packages/ray/tune/examples/ax_example.py hartmann6s, *r#c s`tdD]'}tfddtdD}t|t|t|ddt dqdS)N iterationscs g|] }d|dqS)zx{})getformat).0iconfigr!r" .s z"easy_objective..rr)timesteps_totalr#l2normg{Gz?) rrrrreportr#sqrtsumtimesleep)r+r)rr!r*r"easy_objective,s r4__main__z --smoke-test store_truezFinish quickly for testing)actionhelpzx1 + x2 <= 2.0zl2norm <= 1.25)parameter_constraintsoutcome_constraints)max_concurrentaxr-d)namestopminr2)metricmode search_alg scheduler num_samplesrr)r$x1x2x3x4x5x6) run_config tune_config param_spacez!Best hyperparameters found were: )#__doc__r2numpyrrayrray.tune.schedulersrray.tune.search.axrr#r4__name__argparseArgumentParserparser add_argumentparse_known_argsargs_algosearchConcurrencyLimiterrFTuner RunConfig TuneConfig smoke_testuniformtunerfitresultsprintget_best_resultr+r!r!r!r"s\