Не удалось интерпретировать идентификатор оптимизатора: - PullRequest
0 голосов
/ 18 апреля 2020

Я использую 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'>

1 Ответ

0 голосов
/ 18 апреля 2020

Основная проблема заключается в том, что в

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

, где упоминается оптимизатор, не упоминается скорость обучения. Это должно быть

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

, и теперь оно будет прекрасно работать

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...