首先,我是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]]
我想获得培训中给出的标签名称。知道我怎么能这样做吗?
由于