如何在 Keras LSTM 中查看/更改学习率?

数据挖掘 喀拉斯 lstm 学习率
2022-02-24 21:09:25

我在一些要求降低学习率的问题/答案中看到。但我不知道如何查看和更改库中LSTM模型的学习率Keras

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。你把它改成你想要的任何东西。

有关更多详细信息,请参阅此链接

请告诉我这是否有助于解决您的问题。