«KeyError» возникает при использовании функции прогнозирования в keras - PullRequest
Новая
       101

«KeyError» возникает при использовании функции прогнозирования в keras

0 голосов
/ 10 мая 2019

Я новичок в машинном обучении.Я нашел приведенный ниже код в сети для анализа настроений с использованием набора данных imdb.

С этим набором данных я могу успешно создавать обученную модель.Во время фазы прогнозирования, когда я пытаюсь сделать прогноз для этой модели, получаю некоторый ответ " key error " при отправке строк, которых нет в обучающем наборе .

Где проблема возникает ?.Разве эта модель может только предсказать настроение слов, которые присутствуют в наборе данных?Ваша помощь будет высоко оценена. 

from keras.preprocessing.text import text_to_word_sequence

review = "i love stackoverflow"
text = text_to_word_sequence(review)
tmp = []

for word in text:
    tmp.append(word_to_id[word])
    tmp_padded = sequence.pad_sequences([tmp], maxlen=max_review_length)

print("Sentiment score: %s" % (model.predict(array([tmp_padded][0]))[0][0]))

KeyError Traceback (последний последний вызов)

KeyError: 'stackoverflow'

Код ссылки:https://www.liip.ch/en/blog/sentiment-detection-with-keras-word-embeddings-and-lstm-deep-learning-networks

...