CommencerCommencer gratuitement

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

Afficher le cours

Instructions

  • Journalisez la variable r2_score en 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")
Modifier et exécuter le code