Керас предсказывает Python
Прежде всего, я новичок в Keras и ML.
Я пытаюсь https://github.com/fchollet/keras/blob/master/examples/pretrained_word_embeddings.py пример.
Я сохранил обученную модель, используя model.save(), Сейчас я пишу сценарий для прогнозирования с использованием другого сценария:
model = load_model('./model.h5')
tokenizer = Tokenizer(num_words=MAX_NB_WORDS)
text = np.array(['some unseen text'])
tokenizer.fit_on_texts(text)
sequences = tokenizer.texts_to_sequences(text)
data = pad_sequences(sequences, maxlen=MAX_SEQUENCE_LENGTH)
prediction = model.predict(np.array(data))
Выше приведено что-то вроде:
[[ 1.00000000e+00 2.66153551e-18 1.67506186e-15 1.33851482e-16
2.85387820e-16 3.06079675e-15 1.34407281e-14 1.38863788e-15
4.20352726e-16 1.28687439e-15 1.52688925e-16]]
Я хотел бы получить названия лейблов, приведенные на тренинге. Есть идеи, как я могу это сделать?
Спасибо