Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

save and load a machine learning model using Pickle

#Save the model using pickle
import pickle
# save the model to disk
pickle.dump(model, open(model_file_path, 'wb'))

#Load the model 
model = pickle.load(open(model_file_path, 'rb'))

#Saving a Keras model
# Calling `save('my_model')` creates a SavedModel folder `my_model`.
model.save("my_model")
#Load a Keras Model
# It can be used to reconstruct the model identically.
reconstructed_model = keras.models.load_model("my_model")
Source by www.analyticsvidhya.com #
 
PREVIOUS NEXT
Tagged: #save #load #machine #learning #model #Pickle
ADD COMMENT
Topic
Name
1+1 =