MNIST-heatmap-tensorflow

A script that generates heatmap for all the digits in MNIST dataset

example result

steps to run

1. Install requirements

   $ pip install -r requirements.txt

2. Run the Script

$ python mnist_heatmap.py

  optional arguments:
     -h, --help            show this help message and exit
     --data_dir DATA_DIR   Directory for storing input data
     --batch_size BATCH_SIZE
                           Batch size
     --train_steps TRAIN_STEPS
                         Training Steps