首页 > 解决方案 > 模型编译部分中的 RMSprop 优化器,在 keras 中不起作用

问题描述

当我想编译我的模型时,我使用 tensorflow 2.1.0 和 keras 2.2.4tf [这是我使用的一段代码]:

model.compile(loss='binary_crossentropy',
              optimizer=optimizers.RMSprop(lr=2e-5),
              metrics=['acc'])

它给出了这个错误:

NameError                                 Traceback (most recent call last)
<ipython-input-11-09848482494e> in <module>
     30 
     31 model.compile(loss='binary_crossentropy',
---> 32               optimizer=optimizers.RMSprop(lr=2e-5),
     33               metrics=['acc'])
     34 

NameError: name 'optimizers' is not defined

但是,我不知道是什么问题。如果你能帮助我。

标签: pythontensorflowkerastf.keras

解决方案


尝试调用:

model.compile(loss='binary_crossentropy',
  optimizer=tf.keras.optimizers.RMSprop(lr=2e-5),
  metrics=['acc'])

推荐阅读