我想创建一个迭代模型测试器,但我只发现了变量更改主题。我想更改优化器如下
optimizer_list = ['Adagrad', 'SGD', 'Adam', ..]
for i in model_list:
model.compile(loss='categorical_crossentropy',
optimizer = Adagrad(),
metrics = ['accuracy'])
它没有按我预期的那样工作。进行此操作的正确方法是什么?
提前致谢。
最佳答案
样品溶液
from keras import optimizers
model_list = [optimizers.Adagrad, optimizers.SGD, optimizers.Adam, ..]
for optimization_method in model_list:
model.compile(loss='categorical_crossentropy',
optimizer = optimization_method(),
metrics = ['accuracy'])
如果我理解正确,则您希望维护相同的体系结构,但尝试不同的优化。
您所拥有的问题是,即使您遍历优化策略的名称,也不会1)实际使用它们,2)它们只是字符串,而不是函数
关于python - 循环更改模型(keras,python),我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/53924290/