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
Instructions
- Enregistrez la variable r2_scoresous 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")