【发布时间】:2021-04-27 15:26:16
【问题描述】:
如何使用 Keras Tuner 调整优化功能?我想试试 SGD、Adam 和 RMSprop。
我试过了:
hp_lr = hp.Choice('learning_rate', values=[1e-2, 1e-3, 1e-4])
hp_optimizer = hp.Choice('optimizer', values=[SGD(learning_rate=hp_lr), RMSprop(learning_rate=hp_lr), Adam(learning_rate=hp_lr)])
model.compile(optimizer=hp_optimizer,
loss="sparse_categorical_crossentropy",
metrics=["accuracy"])
但这不起作用,因为“A Choice 只能包含一种类型的值”
【问题讨论】:
标签: python keras deep-learning keras-tuner