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