Tools for Optuna, MLflow and the integration of both
HPOflow Tools for Optuna, MLflow and the integration of both. The main components are: hpoflow.OptunaMLflow: A wrapper to use Optuna and log to MLflow at the same time. hpoflow.OptunaMLflowCallback: Class inheriting from transformers.TrainerCallback that integrates with OptunaMLflowto send the logs to MLflow and Optuna during model training. hpoflow.SignificanceRepeatedTrainingPruner: An Optuna prunerto use statistical significance (a t-test which serves as a heuristic) to stopunpromising trials early, avoiding unnecessary repeated training during cross validation. Installation HPOflow is available at the Python Package […]
Read more