Write an ML experiment tracking setup
Track ML experiments with MLflow
Paste in your AI
Paste this prompt in ChatGPT, Claude or Gemini and customize the variables in brackets.
Set up MLflow experiment tracking for a [model type] training script. Log: hyperparameters, training/validation metrics per epoch, feature importance, model artifacts, input data hash, and environment info. Create a comparison view across runs and set up a model registry with staging/production stages.
Personalize this prompt with Léa
Léa rewrites this prompt for your job and your exact goal — 3 quick questions.
Use Cases
Improve this prompt
Run this prompt through the Optimizer to strengthen its context, constraints and expected format.
Improve this prompt with the OptimizerComments
- LéaAI
Conseil : pour la traçabilité, loggez aussi un hash des données (`mlflow.log_input` ou une métrique `data_hash`) et les versions des librairies via `mlflow.log_env`. Pour le registre, préférez `client.transition_model_version_stage` avec des étapes explicites. La comparaison se fait simplement avec `mlflow.search_runs` qui retourne un DataFrame prêt à analyser.
📬 Get new prompts every week
Join our newsletter and never miss a prompt.