text_plain_output_1.png
import matplotlib.pyplot as plt

x_train = x_train.astype('float32') / 255
y_train = y_train.astype('float32')
x_test = x_test.astype('float32') / 255

plt.figure(figsize=(8, 8))
for i in range(10):
    plt.subplot(5, 5, i + 1)
    plt.title(i)
    plt.imshow(x_train[i].reshape(32, 32, 3))