Inizia subitoInizia gratis

Registrazione di una corsa

In questo esercizio, addestrerai un modello utilizzando Linear Regression di scikit-learn per prevedere i profitti dal dataset Unicorn. Hai creato un esperimento chiamato Unicorn Sklearn Experiment e hai iniziato una nuova sessione. Registrerai le metriche per r2_score e i parametri per n_jobs e registrerai il codice di addestramento come artefatto.

Il modello Linear Regression sarà addestrato con il parametro n_jobs impostato su 1. La metrica r2_score è stata prodotta utilizzando la metrica r2_score() di scikit-learn basata sulla variabile y_pred che deriva dalle previsioni di 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)

Verranno importati il modulo mlflow e i moduli LinearRegression, train_test_split e metrics di scikit-learn.

Questo esercizio fa parte del corso

Introduzione a MLflow

Visualizza corso

Istruzioni dell'esercizio

  • Registra la variabile r2_score come una metrica chiamata "r2_score".
  • Registra un parametro chiamato "n_jobs" sul server di tracciamento.
  • Registra il file "train.py" come artefatto nell'esecuzione.

esercizio interattivo pratico

Prova questo esercizio completando questo codice di esempio.

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

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

# Log the training code
____.____("train.py")
Modifica ed esegui il codice