Journaliser une exécution
Dans cet exercice, vous allez entraîner un modèle avec Linear Regression de scikit-learn pour prédire le profit à partir du jeu de données Unicorn. Vous avez créé une expérience appelée Unicorn Sklearn Experiment et démarré une nouvelle exécution. Vous allez journaliser la métrique r2_score et le paramètre n_jobs, ainsi que le code d’entraînement en tant qu’artefact.
Le modèle Linear Regression sera entraîné avec le paramètre n_jobs défini à 1. La métrique r2_score a été calculée avec r2_score() de scikit-learn à partir de la variable y_pred, issue des prédictions sur X_test.
model = LinearRegression(n_jobs=1)
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
r2_score = r2_score(y_test, y_pred)
Le module mlflow ainsi que les modules LinearRegression, train_test_split et metrics de scikit-learn seront importés.
Cet exercice fait partie du cours
Introduction à MLflow
Instructions
- Journalisez la variable
r2_scoreen tant que métrique appelée"r2_score". - Journalisez un paramètre appelé
"n_jobs"vers le Tracking Server. - Journalisez le fichier
"train.py"comme artefact dans l’exécution.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Log the metric r2_score as "r2_score"
____.____("____", ____)
# Log parameter n_jobs as "n_jobs"
____.____("____", ____)
# Log the training code
____.____("train.py")