Een run loggen
In deze oefening train je een model met scikit-learns Linear Regression om de winst uit de Unicorn-gegevensset te voorspellen. Je hebt een experiment met de naam Unicorn Sklearn Experiment gemaakt en een nieuwe run gestart. Je gaat metrics voor r2_score loggen en parameters voor n_jobs, en je logt ook de trainingscode als een artefact.
Het Linear Regression-model wordt getraind met de parameter n_jobs ingesteld op 1. De r2_score-metric is berekend met r2_score() uit scikit-learn op basis van de variabele y_pred, die afkomstig is van voorspellingen op 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)
De mlflow-module en de modules LinearRegression, train_test_split en metrics uit scikit-learn worden geïmporteerd.
Deze oefening maakt deel uit van de cursus
Introductie tot MLflow
Oefeninstructies
- Log de variabele
r2_scoreals een metric met de naam"r2_score". - Log een parameter met de naam
"n_jobs"naar de Tracking Server. - Log het bestand
"train.py"als een artefact in de run.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Log the metric r2_score as "r2_score"
____.____("____", ____)
# Log parameter n_jobs as "n_jobs"
____.____("____", ____)
# Log the training code
____.____("train.py")