Spara och läsa in en modell
Med Model API kan modeller delas mellan utvecklare som kanske inte har tillgång till samma MLflow Tracking-server, genom att använda ett lokalt filsystem.
I den här övningen tränar du en ny LinearRegression-modell baserad på en befintlig modell med hjälp av Unicorn-datamängden. Först läser du in en befintlig modell från det lokala filsystemet. Sedan tränar du en ny modell utifrån den befintliga och sparar den tillbaka till det lokala filsystemet.
Den befintliga modellen har sparats till det lokala filsystemet i en katalog som heter "lr_local_v1". Modulen mlflow importeras automatiskt.
Den här övningen är en del av kursen
Introduktion till MLflow
Övningsinstruktioner
- Läs in modellen från den lokala filsystemkatalogen
"lr_local_v1"med hjälp av scikit-learn-biblioteket från MLflow-modulen. - Spara modellen lokalt till en katalog som heter
"lr_local_v2"med hjälp av scikit-learn-biblioteket frånmlflow-modulen.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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
____.____.____(____, "____")