gpt4 book ai didi

python - 在 keras(多类)上训练模型后进行预测?

转载 作者:行者123 更新时间:2023-12-01 08:59:25 25 4
gpt4 key购买 nike

我正在训练一个多类文本分类器,我知道这是一个幼稚的问题,但我对 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(?)

最佳答案

您必须像处理训练数据一样对输入字符串进行tokenizepad_sequences,然后调用model_lstm.predict(tokenized_and_padded_string) 关于它。此链接有一些代码您可以查看:link

关于python - 在 keras(多类)上训练模型后进行预测?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52576017/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com