我在一些要求降低学习率的问题/答案中看到。但我不知道如何查看和更改库中LSTM模型的学习率Keras?
如何在 Keras LSTM 中查看/更改学习率?
数据挖掘
喀拉斯
lstm
学习率
2022-02-24 21:09:25
1个回答
在 Keras 中,您可以将学习率设置为优化方法的参数,下面的代码是 Keras 文档中的示例:
from keras import optimizers
model = Sequential()
model.add(Dense(64, kernel_initializer='uniform', input_shape=(10,)))
model.add(Activation('softmax'))
sgd = optimizers.SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss='mean_squared_error', optimizer=sgd)
在上面的代码中,第五行lr是学习率,在这段代码中设置为 0.01。你把它改成你想要的任何东西。
有关更多详细信息,请参阅此链接。
请告诉我这是否有助于解决您的问题。
其它你可能感兴趣的问题