iT邦幫忙

2023 iThome 鐵人賽

DAY 20
0
AI & Data

機器學習不難嘛系列 第 20

Day20-MNIST基本控制

  • 分享至 

  • xImage
  •  

接下來我們要學會的是一寫MNIST的基本訓練程式碼

首先是察看確認之前提到的訓練資料和測試資料分別為60000筆盒10000筆,相信有學過一點點Python的人都知道查看長度是用{len(資料)},道理是一樣的,我這裡直接進行示範:

print(len(train_feature),len(train_label))

再來可以確認照片的維度,在Python我們是使用{資料.shape}來查看

print(train_feature.shape,train_label.shape)

這邊的第一個60000和28分別為圖片的真實值和28*28的長寬

還有一個很重要的是印出數字圖片的照片,通常會定義一個show_image函數來印出照片

import matplotlib.pyplot as plt
def show_image(image):
  fig = plt.gcf()
  fig.set_size_inches(2, 2)
  plt.imshow(image, cmap='binary')
  plt.show()

程式解釋:

第1行載入matplotlib.pyplot,並簡稱plt,是為了利用matplotlib來進行繪圖

第2行創建一個函數show_image,並要求輸入一個image參數

第3行設立一個變數fig,gcf()是一個matplotlib中的一個函數,是為了對fig進行操作

第4行將圖片長寬設定為2英寸

第5行的imshow為matplotlib中的一個函數,裡面的第1個參數image為呼叫函數中輸入的值image

,第2個參數cmap=’binary’是用來設定照片為白底黑字的方式顯示的

第6行是將照片用以上條件來顯示出來

直接來印一個試試看吧

show_image(train_feature[0])

https://ithelp.ithome.com.tw/upload/images/20230928/20162311e8WCtw0m4E.png

可以練習看看一次輸入10個image值來印出十張照片,明天教預測資料時可以順便對一下答案。


上一篇
Day19-認識MNIST
下一篇
Day21-MNIST察看結果
系列文
機器學習不難嘛30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言