CommencerCommencer gratuitement

Enregistrement d'une exécution

Dans cet exercice, vous entraînerez un modèle à l'aide de scikit-learn's Linear Regression pour prédire les bénéfices à partir de l'ensemble de données Unicorn. Vous avez créé une expérience appelée Unicorn Sklearn Experiment et commencé un nouveau cycle. Vous enregistrerez les métriques pour r2_score et les paramètres pour n_jobs, ainsi que le code d'entraînement en tant qu'artefact.

Le modèle Linear Regression sera formé avec le paramètre n_jobs fixé à 1. La métrique r2_score a été produite en utilisant la variable r2_score() de scikit-learn basée sur la variable y_pred qui provient des prédictions de 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

  • Enregistrez la variable r2_score sous la forme d'une métrique appelée "r2_score".
  • Enregistrez un paramètre appelé "n_jobs" dans le serveur de cursus.
  • Enregistrez le fichier "train.py" en tant qu'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