o 1 iùã@s<dZddlZddlZddlZddlZddlmZddlmZm Z dd„Z dd„Z dd d „Z d d „Z ddd„ZedkršddlZe ¡Zejddddejdedde ¡\ZZejreej e ¡d¡ZnejZe eejdejs€e e d¡j g¡Z!e"e!ƒeeejdejsœe e d¡j g¡Z!e"e!ƒdSdSdS)z5Examples using MLfowLoggerCallback and setup_mlflow. éN)Útune)ÚMLflowLoggerCallbackÚ setup_mlflowcCsd||dd|dS)Nçš™™™™™¹?édéÿÿÿÿ©)ÚstepÚwidthÚheightrrúl/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/tune/examples/mlflow_example.pyÚ evaluation_fnsr cCsR|d|d}}t| dd¡ƒD]}t|||ƒ}t ||dœ¡t d¡qdS)Nr r Ústepsr©Z iterationsÚ mean_lossr)ÚrangeÚgetr rÚreportÚtimeÚsleep©Úconfigr r r Zintermediate_scorerrr Útrain_functions   ûrFc Cs\tjttjdt|dddgdtjddt dd ¡t d d ¡|r"dnd d œd }| ¡dS) NÚmlflowÚexampleT)Ú tracking_uriÚexperiment_nameZ save_artifact)ÚnameÚ callbacksé©Z num_samplesé rr)r r r©Z run_configZ tune_configZ param_space)rÚTunerrÚ RunConfigrÚ TuneConfigÚrandintÚfit©Úmlflow_tracking_uriÚ finish_fastZtunerrrr Útune_with_callbacks(ýÿþ ÿ   ýñ r+cCsnt|ƒ|d|d}}t| dd¡ƒD]}t|||ƒ}tjt|d|dt ||dœ¡t   d¡qdS) Nr r rr)r)r rr) rrrr rZ log_metricsÚdictrrrrrrrr Útrain_function_mlflow8s  ùr-c Cspt |¡tjddtjttjddtjddt dd¡t d d¡|r&dnddt  ¡d œd œd }|  ¡dS) NÚ mixin_example)rr)rrr r!rr)rr)r r rrr") rZset_tracking_uriZset_experimentrr#r-r$r%r&Zget_tracking_urir'r(rrr Útune_with_setupHs&  ÿÿ   þüø r/Ú__main__z --smoke-testÚ store_truezFinish quickly for testing)ÚactionÚhelpz--tracking-uriz0The tracking URI for the MLflow tracking server.)Útyper3Zmlruns)r*rr.)F)#Ú__doc__ÚosÚtempfilerrZrayrZray.air.integrations.mlflowrrr rr+r-r/Ú__name__ÚargparseÚArgumentParserÚparserÚ add_argumentÚstrÚparse_known_argsÚargsÚ_Z smoke_testÚpathÚjoinÚ gettempdirr)rZ search_runsZget_experiment_by_nameZ experiment_idÚdfÚprintrrrr ÚsR   ÿý  ÿ ÿ á