python - 无法解释优化器标识符：

Question

我正在使用 Pure Keras 制作一些模型，这就是它。

from keras import models,layers,Sequential, losses,metrics,optimizers
from keras.datasets import imdb
(train_data,train_labels),(test_data,test_labels) = imdb.load_data(num_words=10000)
model = models.Sequential()
model.add(layers.Dense(16,activation='relu',input_shape=(10000,)))
model.add(layers.Dense(16,activation='relu'))
model.add(layers.Dense(1,activation='sigmoid'))
model.compile(optimizer = optimizers.rmsprop, loss = losses.binary_crossentropy,metrics=['accuracy'])
model.fit(x_train,y_train,epochs=5,batch_size=512)
results = model.evaluate(x_test,y_test)

我得到的错误是这个

Could not interpret optimizer identifier: <class 'keras.optimizers.RMSprop'>

标签： pythonkerasdeep-learning

Answer 1

主要问题是在

model.compile(optimizer = optimizers.rmsprop, loss = losses.binary_crossentropy,metrics=['accuracy'])

提到优化器，没有提到学习率。它应该是

model.compile(optimizer = optimizers.rmsprop(0.01), loss = losses.binary_crossentropy,metrics=['accuracy'])

现在它可以正常工作了