keras怎么设置/改动rmsprop的学习率?

  统计/机器学习 深度学习 Python 人工神经网络    浏览次数:1545        分享
0

keras里训练前要compile一下并且设置优化算法,比如用rmsprop算法就是下面这样

model.compile(optimizer='rmsprop',
                loss='categorical_crossentropy',
                metrics=['accuracy'])

我的问题是我怎么在keras里设置或者改动默认的学习率?

 

offer雨   2019-05-22 14:14



   1个回答 
4


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



  相关主题

keras.models.Sequential()是一种什么模型   1回答

keras里的predict,predict_classes,predict_proba有什么区别?   2回答

keras hdf5 從某層輸入數值   1回答

为什么Keras和Pytorch都是把模型保存成h5的格式而不是pkl格式?   2回答

keras和tensorflow有什么关系?   3回答

keras load_model报错   1回答

keras sequential model怎么增加一个dropout层?   1回答

keras加载imdb数据时num_words参数是干嘛的?   1回答

怎么在keras中加载MNSIT数据集?   2回答

训练完keras模型后怎么画出loss和val_loss的对比图?   1回答

keras无法加载imdb文本:Object arrays cannot be loaded when allow_pickle=False   1回答

keras模型合并请教   2回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!