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

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

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

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

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

 

offer雨   2019-05-22 14:14



   1个回答 
5


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()是一种什么模型

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

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

keras load_model报错

keras和tensorflow有什么关系?

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

keras hdf5 從某層輸入數值

Keras模型针对同一数据集做fit和evaluation,accuracy值却不相同

keras模型合并请教

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

  随便看看

如何获取pyspark DataFrame的行数和列数?

凸函数、凸集分别是什么意思?

pandas同时返回一个dataframe的前几行(head)和后几行(tail)

对连续特征一定要进行分箱处理吗?

seaborn如何显示图?