我正在训练一个多类文本分类器,我知道它是天真的问题,但是我对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(?)
答案 0 :(得分:1)
您必须像对待训练数据一样在输入字符串上tokenize
和pad_sequences
,然后在其上调用model_lstm.predict(tokenized_and_padded_string)
。
该链接包含一些代码,您可以查看:link