logo
Loading...

Dense參數的units和input_dim意思? - Cupoy

model.add(Dense(units=256,               input_dim...

ml100-2,ml100-2-d70

Dense參數的units和input_dim意思?

2019/07/15 下午 06:26
機器學習共學討論版
施渝陽
觀看數:470
回答數:1
收藏數:0
ml100-2
ml100-2-d70

model.add(Dense(units=256, 

              input_dim=784, 

              kernel_initializer='normal', 

              activation='relu'))

老師您好,因為這地方對於Dense裡的參數不是很理解,因此想請問您裏頭的units和input_dim是甚麼意思,哪裡不相同?

是不是一個指輸入的數目,另一個是指隱藏層的層數呢?謝謝。

回答列表

  • 2019/07/15 下午 11:38
    Jimmy
    贊同數:1
    不贊同數:0
    留言數:3

    Hi 渝陽!


    沒錯喔! input_dim 指的就是 input dimension 的意思,讓這層 Dense layer 知道你 inputs 的維度是多少,接下來再透過妳給定的 units 來決定 output 的維度囉