test.py

# decoding the labels
predicted_labels = decoder(model.predict(test_images))
test_labels = decoder(test_labels)

row = 3
col = 4
fig, axes = plt.subplots(row, col, figsize=(16, 12))
c = 0
count = 0
for i in range(row):
    for j in range(col):
        axes[i][j].imshow(test_images[c])
        axes[i][j].set_title(f'Predicted: {predicted_labels[c]}', fontsize=14)
        axes[i][j].set_xlabel(f'Actual: {test_labels[c]}', fontsize=14)
        if (predicted_labels[c] != test_labels[c]):
            count+=1
        c += 80
        
plt.tight_layout()