keras里训练前要compile一下并且设置优化算法，比如用rmsprop算法就是下面这样
model.compile(optimizer='rmsprop',
loss='categorical_crossentropy',
metrics=['accuracy'])我的问题是我怎么在keras里设置或者改动默认的学习率？
1个回答
from keras import optimizers
rms = optimizers.RMSprop(lr=0.001, rho=0.9, epsilon=None, decay=0.0)
model.compile(optimizer=rms,
loss='categorical_crossentropy',
metrics=['accuracy'])修改lr即可
SofaSofa数据科学社区DS面试题库 DS面经
zlsama
2019-05-22 15:12
相关讨论
随便看看