Answers for "save and reload model in python"

3

save machine learning model

# fit the model
model.fit(X_train, y_train)

# save the model
import pickle
pickle.dump(model, open("model.pkl", "wb"))

# load the model
model = pickle.load(open("model.pkl", "rb"))

# use model to predict
y_pred = model.predict(X_input)
Posted by: Guest on July-24-2020
5

load model keras

from tensorflow import keras
model = keras.models.load_model('path/to/location')
Posted by: Guest on June-04-2020

Python Answers by Framework

Browse Popular Code Answers by Language