Ein Modell speichern und laden
Mit der Model API können Modelle über das lokale Dateisystem zwischen Entwicklern geteilt werden, die eventuell keinen Zugriff auf denselben MLflow-Tracking-Server haben.
In dieser Übung trainierst du ein neues LinearRegression-Modell aus einem bestehenden Modell mit dem Datensatz Unicorn. Zuerst lädst du ein vorhandenes Modell aus dem lokalen Dateisystem. Dann trainierst du aus dem bestehenden Modell ein neues und speicherst es wieder im lokalen Dateisystem.
Das bestehende Modell wurde im lokalen Dateisystem in einem Verzeichnis namens "lr_local_v1" gespeichert. Das Modul mlflow wird importiert.
Diese Übung ist Teil des Kurses
Einführung in MLflow
Anleitung zur Übung
- Lade das Modell aus dem lokalen Verzeichnis
"lr_local_v1"mithilfe der scikit-learn-Bibliothek aus dem MLflow-Modul. - Verwende die scikit-learn-Bibliothek aus dem
mlflow-Modul, um das Modell lokal in ein Verzeichnis namens"lr_local_v2"zu speichern.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Load model from local filesystem
model = ____.____.____("____")
# Training Data
X = df[["R&D Spend", "Administration", "Marketing Spend", "State"]]
y = df[["Profit"]]
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.7,random_state=0)
# Train Model
model.fit(X_train, y_train)
# Save model to local filesystem
____.____.____(____, "____")