LoslegenKostenlos loslegen

Einen Run protokollieren

In dieser Übung trainierst du mit scikit-learns Linear Regression ein Modell, das den Profit aus dem Unicorn-Datensatz vorhersagt. Du hast ein Experiment namens Unicorn Sklearn Experiment erstellt und einen neuen Run gestartet. Du wirst Metriken für r2_score und Parameter für n_jobs protokollieren sowie den Trainingscode als Artefakt ablegen.

Das Linear Regression-Modell wird mit dem Parameter n_jobs auf 1 trainiert. Die Metrik r2_score wurde mit r2_score() aus scikit-learn auf Basis der Variable y_pred berechnet, die aus Vorhersagen für X_test stammt.

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)

Das Modul mlflow sowie die Module LinearRegression, train_test_split und metrics aus scikit-learn werden importiert.

Diese Übung ist Teil des Kurses

Einführung in MLflow

Kurs anzeigen

Anleitung zur Übung

  • Protokolliere die Variable r2_score als Metrik mit dem Namen "r2_score".
  • Protokolliere einen Parameter namens "n_jobs" beim Tracking-Server.
  • Protokolliere die Datei "train.py" als Artefakt im Run.

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# Log the metric r2_score as "r2_score"
____.____("____", ____)

# Log parameter n_jobs as "n_jobs"
____.____("____", ____)

# Log the training code
____.____("train.py")
Code bearbeiten und ausführen