Keras模型编译设置(查看/更改)



假设我有以下Keras模型:

model = Sequential()
model.add(Dense(units=64, activation='relu'))
model.add(Dense(units=10, activation='softmax'))
model.compile(
loss      = CategoricalCrossentropy(label_smoothing=0.01),
optimizer = RMSprop(learning_rate=0.001, momentum=0.0)
metrics   = [Accuracy()]
)

我有两个问题:

  1. 如何查看编译设置(如learning_rate)?
  2. 如何更改编译设置(如learning_rate)?

备注:

  • 我注意到我可以使用model.summary()或model.get_config()查看图层设置,但这并不显示编译设置。
  • 我知道我可以用不同的learning_rate再次运行编译语句来改变learning_rate。但我想要一个"更干净"/更可读的方式来做到这一点。比如:model['compilation']['optimizer']['learning_rate'] = xxx。(许多sklearn模型可以通过这种方式进行调整)

使用.lr:

rate = model.optimizer.lr

相关内容

  • 没有找到相关文章

最新更新