在構建輸出層的時候 Dense的units 為什麼是10 ，而不是256? - Cupoy

2019/07/25 下午 03:17

機器學習共學討論版

luohq_dsr

觀看數：50

回答數：2

收藏數：0

ml100-2

ml100-2-d70

回答列表

• 2019/07/25 下午 06:13
  張維元 (WeiYuan)
  贊同數：0
  不贊同數：0
  留言數：0

  嗨，可以參考這一篇的討論：https://www.cupoy.com/qa/kwassist/ai_tw/0000016BCB719949000000476375706F795F72656C656173655155455354

  
  

  輸出層會等於你的資料類別？因為手寫辨識的資料是 0 ~ 9 有 10 種可能。

• 2019/07/26 下午 11:34
  Jimmy
  贊同數：2
  不贊同數：0
  留言數：0

  Hi luohq_dsr!

  
  

  我們在建構神經網路時，要指定網路的輸入與輸出的維度，輸入的維度是影像尺寸，因此就是 28x28=784，然後接上 256 個神經元 (units) ，至於模型的輸出，因為是手寫數字辨識，共有十個數字，所以我們在模型的最後接上 10 個神經元 (units)，再經過 softmax 函數把數值壓逤到 0~1 之間，得到 10 個數字各自的預測機率值。

  
  

  所以模型的輸入和輸出都是根據你的資料而定的! 當你今天是要做兩個類別的分類，模型的輸出 units 就會改成 2  囉!