TensorFlow with Keras part 2 using Case Study

Reading Time: 2 minutes

This blog is in continuation to Part1, Tensorflow for deep learning. Make sure you go through it for a better understanding of this case study.

Keras is a high-level neural networks API, written in Python and capable of running on top of Tensorflow, CNTK  or Theano. It was developed with a focus on enabling fast experimentation. In this blog, we are going to cover one small case study for fashion mnist.

Fashion-MNIST is a dataset of Zalando’s article images—consisting of a training set of 60,000 examples and a test set of 10,000 examples. Each example is a 28×28 grayscale image, associated with a label from 10 classes. Zalando intends Fashion-MNIST to serve as a direct drop-in replacement for the original MNIST dataset for benchmarking machine learning algorithms. It shares the same image size and structure of training and testing splits.


Each training and test example is assigned to one of the following labels:

  • 0 T-shirt/top
  • 1 Trouser
  • 2 Pullover
  • 3 Dress
  • 4 Coat
  • 5 Sandal
  • 6 Shirt
  • 7 Sneaker
  • 8 Bag
  • 9 Ankle boot

1)  First Load Required packages

import tensorflow as tf
from tensorflow import keras
import numpy as np
import matplotlib.pyplot as plt

2) Load the dataset

which will be downloaded the data set to your system

fashon_mnist = keras.datasets.fashion_mnist
(train_images,train_labeks),(test_images,test_labels) = fashon_mnist.load_data(

3) Visualization for the first few rows

Let’s plot some samples for the images. We add labels to the train set images, with the corresponding fashion item category.

for i in range(10):

train_images = test_images/255.0
test_images = test_images/255.0

class_name = ['T-shirt/top','Trouser','Pullovers','Dress','Coat','Sandals','Shirt','Sneaker','Bag','Ankle boot']

for i in range(25):

4) Model Creation

We start by preparing the model. We will use a Sequential model. The Sequential model is a linear stack of layers. It can be first initialized and then we add layers using add method or we can add all layers at the init stage. The layers added are as follows: Dense. This layer is a regular fully-connected NN layer. It is used without parameters:units: – this is a positive integer, with the meaning: dimensionality of the output space; in this case is: 128;activation – activation function : relu;

Dense. This is the final layer (fully connected). It is used with the parameters:units: the number of classes (in our case 10);activation : softmax; for this final layer it is used softmax activation (standard for multiclass classification) units – this is a positive integer, with the meaning: dimensionality of the output space; in this case is: 128;activation – activation function : relu;

model = keras.Sequential(

5) Compiling model

Then we compile the model, specifying as well the following parameters:

  • loss;
  • optimizer;
  • metrics.

6) Fit the model


7) Test Prediction Accuracy

We calculate the test loss and accuracy.

test_loss , test_acc = model.evaluate(test_images,test_labels)
#Test accuracy is around 0.81

8) Prediction

Now we can use the trained model to make predictions/classifications on the test dataset model.predict(x_test)

prediction = model.predict(test_images)
for i in range(10): 
    print("expected -",class_name[test_labels[i]])
    print("predicted-" ,class_name[np.argmax(prediction[i])])


  1. https://www.tensorflow.org/
  2. https://www.kaggle.com

1 thought on “TensorFlow with Keras part 2 using Case Study3 min read

Comments are closed.