[英]how to visualize prediction classes in keras?
我想可視化經過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)
28 * 28!=3072。我假設每個圖像尺寸為(32,32,3)(32 * 32 * 3 == 3072)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.