как визуализировать классы предсказания в керасе?

Question

Я хочу визуализировать результаты прогнозирования из моей обученной модели cnn. Но я сталкиваюсь с ошибкой: не могу преобразовать массив размером 3072 в форму (28,28).Код указан ниже:

L = 5
W = 5
fig, axes = plt.subplots(L, W, figsize = (12,12))
axes = axes.ravel() # 

for i in np.arange(0, L * W):  
    axes[i].imshow(x_test[i].reshape(28,28))
    axes[i].set_title("Prediction Class = {:0.1f}\n True Class 
    {:0.1f}".format(predicted_classes[i], y_test[i]))
    axes[i].axis('off')

plt.subplots_adjust(wspace=0.5)

Answer 1

28 * 28! = 3072. Я предполагаю, что размеры каждого изображения (32,32,3) (32 * 32 * 3 == 3072)