Aan de slagGa gratis aan de slag

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

Cursus bekijken

Oefeninstructies

  • Log de variabele r2_score als 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")
Code bewerken en uitvoeren