In [1]:

    

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
import tensorflow as tf
import matplotlib.pyplot as plt
import numpy as np

(X_train, y_train), (X_test, y_test) = tf.keras.datasets.mnist.load_data()


    

In [ ]:

    

model = tf.keras.models.load_model("perceptron.h5")
is_five_test = False
image = X_test[y_test != 5][10]
pred = model.predict(np.array([image]))[0][0]
plt.axis("off")
plt.title("label {}".format(pred))
plt.imshow(image, cmap='gray', interpolation='none')


    

In [ ]: