在上节中编译模型时,直接使用名称来配置:
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['acc'] )
也可以使用keras的Adam优化器实例:
model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.01,
loss='sparse_categorical_crossentropy',
metrics=['acc']
)
learing_rate是学习速率,默认0.001,设置为0.01时,会发现使用这个较大的学习速率后,模型训练的结果不是非常好。