Save and Load Your Keras Deep Learning Model

In this tutorial, we will learn about Keras and how to use it i.e. how to create a deep learning model on Keras, how to save it, and how to load it. So, let’s first start with Keras.

Keras is basically a library for simplifying deep learning. It works with TensorFlow in the backend and was developed by Microsoft. It is easier than Tensorflow but is slower than TensorFlow. This is the reason for its use in small models, with large models TensorFlow is preferred.

How to save a model using Keras:

Importing Keras for model creation:

Use the following code for importing Keras:

from keras.models import Sequential
from keras.layers import Dense
import numpy as np

Notice that it will show “Using Tensorflow backend”, this is because Keras works on the TensorFlow backend of Python.

Creating Model:

Here, we will create a model. We will first create a model and put sequential (imported from keras.models) in it. Then we will add hidden layers to our model according to our convenience. Here, we have created three hidden layers. The first hidden layer will have 32 neurons and will take 10 inputs. In the second hidden layer, there are 10 neurons and now the activation function used is “relu”. Similarly in the last layer, there are 5 neurons and the activation function used is “sigmoid”.

In the end, the whole neural network will compile. Here, we will give optimizer i.e. used to give weight to the layers and loss which is given to minimize the cost of the network.

See the Python code below:

model = Sequential()
model.add(Dense(32, input_dim=10))
model.add(Dense(10, activation="relu"))
model.add(Dense(5, activation="sigmoid"))
model.compile(optimizer="adam",loss="mse")

Saving Model:

Here, we will save the model using model.save command. In double quotes give the location where you want your model to be saved.

model.save('/content/drive/My Drive/Colab Notebooks/simple.h5')

How to load model using Keras:

Now, to load the previously saved model use the following code. In quotes provide the path where you have saved your model previously.

To use the model we will give the inputs for the first layer in “model.predict” function. Here, we have given 10 zeroes as input.

from keras.models import load_model
import numpy as np

model=load_model("/content/drive/My Drive/Colab Notebooks/simple.h5")

print(model.summary())
test=np.expand_dims(np.zeros(10), axis=0)
print(model.predict(test))

 

Output:

Model: "sequential_1" 
_________________________________________________________________ 
Layer (type) Output Shape Param # 
================================================================= 
dense_1 (Dense) (None, 32) 352 
_________________________________________________________________ 
dense_2 (Dense) (None, 10) 330 
_________________________________________________________________ 
dense_3 (Dense) (None, 5) 55 
================================================================= 
Total params: 737 
Trainable params: 737 
Non-trainable params: 0 
_________________________________________________________________ 
None [[0.5 0.5 0.5 0.5 0.5]]

Leave a Reply

Your email address will not be published.