iT邦幫忙

第 11 屆 iThome 鐵人賽

DAY 17
0
自我挑戰組

Tensorflow學習日記系列 第 17

tensorflow學習日記Day17 多元感知器模型 增加神經元

  • 分享至 

  • xImage
  •  

#把隱藏層的神經元個數從256調整為1000
model = Sequential()
model.add(Dense(units=1000,input_dim=784,kernel_initializer='normal',activation='relu'))
model.add(Dense(units=10,kernel_initializer='normal',activation='softmax'))
#輸出模型摘要
print(model.summary())
#開始訓練
model.compile(loss='categorical_crossentropy',optimizer='adam',metrics=['accuracy'])
train_history=model.fit(x=x_Train_normalize,y=y_Train_OneHot,validation_split=0.2,epochs=10,batch_size=200,verbose=2)
訓練結果:
https://scontent.ftpe8-1.fna.fbcdn.net/v/t1.15752-9/71402157_533595213879832_1378952469320564736_n.png?_nc_cat=105&_nc_oc=AQlbpic8e33MDsyTPa3SyfSIG-UEmzjkeLJwWeHBV4x0QV_Tv4TLGH-8opmqRed2B3I&_nc_ht=scontent.ftpe8-1.fna&oh=8dc816174120cdd1121c306e81b547b9&oe=5E2C7B9C

#查看準確率
show_train_history(train_history,'accuracy','val_accuracy')
https://scontent.ftpe8-4.fna.fbcdn.net/v/t1.15752-9/72267645_2540606609548771_4184713239243784192_n.png?_nc_cat=110&_nc_oc=AQkLJF3ym1GIT3R8g8CDWuk_gKp8xZsM2uG8J8sjb5EIsEo-GQEhm209kFyJyaeeLLQ&_nc_ht=scontent.ftpe8-4.fna&oh=afd4b3a14be59d92479ca4cff0a3306a&oe=5E2A6040
#預測確率
scores = model.evaluate(x_Test_normalize,y_Test_OneHot)
print()
print('accuracy=',scores[1])
輸出:
10000/10000 [==============================] - 1s 69us/step
accuracy= 0.9785000085830688


上一篇
tensorflow學習日記Day16 多元感知器模型預測(下)
下一篇
tensorflow學習日記Day18 Overfitting和DropOut
系列文
Tensorflow學習日記30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言