o 1 iú ã@s<dZddlZddlZddlmZddlmZddlm Z dd„Z dd „Z e d krœddl Z e  ¡Zejd d d de ¡\ZZe dgdgdZejjeddZeƒZeje ejdddidejddeeejrgdnddde dd¡e dd¡e dd¡e dd¡e dd¡e dd¡dœdZe ¡Ze d e !¡j"ƒdSdS)!zÂThis 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 Cs¾t gd¢¡}t gd¢gd¢gd¢gd¢g¡}dt gd¢gd¢gd ¢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-Cëâ6?)i i iÁé|i[ iþ)i i'is i˜iìi')i, i«i iC iç iú)iÏi|"i"ioiCi}çréé)ÚnpÚarrayÚ enumerateÚrangeÚexp) ÚxÚalphaÚAÚPÚyÚjZalpha_jÚtÚk©r úh/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/tune/examples/ax_example.pyÚ hartmann6s,üÿüÿ *r"c s`tˆdƒD]'}t ‡fdd„tdƒDƒ¡}t |t|ƒt |d ¡¡dœ¡t  d¡qdS)NÚ iterationscs g|] }ˆ d |d¡¡‘qS)zx{}é)ÚgetÚformat)Ú.0Úi©Úconfigr r!Ú .s z"easy_objective..rr)Útimesteps_totalr"Zl2normg{®Gáz”?) rrrrÚreportr"ÚsqrtÚsumÚtimeÚsleep)r*r(rr r)r!Úeasy_objective,sýÿ ÷r2Ú__main__z --smoke-testÚ store_truezFinish quickly for testing)ÚactionÚhelpzx1 + x2 <= 2.0zl2norm <= 1.25)Zparameter_constraintsZoutcome_constraintsé)Zmax_concurrentÚaxr,éd)ÚnameÚstopÚminré2)ZmetricÚmodeZ search_algÚ schedulerZ num_samplesrr)r#Úx1Zx2Zx3Zx4Zx5Zx6)Z run_configZ tune_configZ param_spacez!Best hyperparameters found were: )#Ú__doc__r0ÚnumpyrZrayrZray.tune.schedulersrZray.tune.search.axrr"r2Ú__name__ÚargparseÚArgumentParserÚparserÚ add_argumentÚparse_known_argsÚargsÚ_ÚalgoÚsearchZConcurrencyLimiterr?ZTunerZ RunConfigZ TuneConfigZ smoke_testÚuniformZtunerÚfitÚresultsÚprintZget_best_resultr*r r r r!Ús\    ÿ þþ û      ùóØ