在对keras(Multiclass)训练模型之后进行预测吗?

时间:2018-09-30 08:12:54

标签: python keras

我正在训练一个多类文本分类器,我知道它是天真的问题,但是我对ML天真。

所以我导入了所有加载的数据集,看起来像这样

label  post
0      Don't worry
1      When did I?
2      By by 

然后我将句子标记化 并定义了模型

然后训练它

model.fit(data, labels, validation_split=0.2, epochs=5)

现在如何对自定义文本输入进行预测?

model_lstm.predict(?)

1 个答案:

答案 0 :(得分:1)

您必须像对待训练数据一样在输入字符串上tokenizepad_sequences,然后在其上调用model_lstm.predict(tokenized_and_padded_string)。 该链接包含一些代码,您可以查看:link