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
Istruzioni dell'esercizio
- Registra la variabile
r2_scorecome 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")