1import matplotlib.pyplot as plt
 2import numpy as np
 3from random import randint
 4
 5
 6
 7dataset = np.load('dataset.npy')
 8
 9n = randint(0, len(dataset))
10
11plt.plot(111)
12plt.axis('off')
13plt.imshow(dataset[n][0])
14plt.title('class: ' + str(dataset[n][1]))
15
16plt.show()