|
感謝您的訂正: 請將第 26、28(兩處)、29、32 原來的 i 改為 start_id,如下:
15 def show_images_labels_predictions(images,labels, 16 predictions,start_id,num=10): 17 plt.gcf().set_size_inches(12, 14) 18 if num>25: num=25 19 for i in range(0, num): 20 ax=plt.subplot(5,5, 1+i) 21 #顯示黑白圖片 22 ax.imshow(images[start_id], cmap='binary') 23 24 # 有 AI 預測結果資料, 才在標題顯示預測結果 25 if( len(predictions) > 0 ) : 26 title = 'ai = ' + str(predictions[start_id]) 27 # 預測正確顯示(o), 錯誤顯示(x) 28 title += (' (o)' if predictions[start_id]==labels[start_id] else ' (x)') 29 title += '\nlabel = ' + str(labels[start_id]) 30 # 沒有 AI 預測結果資料, 只在標題顯示真實數值 31 else : 32 title = 'label = ' + str(labels[start_id]) 33 34 # X, Y 軸不顯示刻度 35 ax.set_title(title,fontsize=12) 36 ax.set_xticks([]);ax.set_yticks([]) 37 start_id+=1 38 plt.show() |