Python | Classify handwritten numbers with Tensorflow

# importing libraries

import numpy as np

import matplotlib.pyplot as plt

import tensorflow as tf

learn = tf.contrib.learn

tf.logging.set_verbosity (tf.logging.ERROR)

 
# importing the dataset using MNIST
# this is how mnist is used mnist contains the dataset test and train

mnist = learn. dat asets.load_dataset ( ’mnist’ )

data = mnist.train.images

labels = np.asarray (mnist.train.labels, dtype = np.int32)

test_data = mnist.test.images

test_labels = np.asarray (mnist.test.labels, dtype = np. int32)

max_examples = 10000

data = data [: max_examp les]

labels = labels [: max_examples]

 
# displaying the dataset with Matplotlib

def display (i):

img = test_data [i]

plt.title ( ’label: {}’ . format (test_labels [i]))

plt.imshow (img.reshape (( 28 , 28 )))

 
# img in tf - 28 by 28 px
# approximate linear classifier

feature_columns = learn.infer_real_valued_columns_from_input (data)

classifier = learn.LinearClassifier (n_classes = 10 , 

feature_columns = feature_columns)

classifier.fit (data, labels, batch_size = 100 , steps = 1000 )

 
# Evaluate the accuracy
classifier.evaluate (test_data, test_labels)

print (classifier.evaluate (test_data, test_labels) [ "accuracy" ] )

prediction = classifier.predict (np.array ([test_data [ 0 ]], 

  dtype = float ), 

as_iterable = False )

print ( "prediction: {} , label : {} " . format (prediction, 

test_labels [ 0 ]))

if prediction = = test_labels [ 0 ]:

  display ( 0 )