nlp - 如何使用 DeepPavlov 进行文本分类

Question

我有兴趣使用 DeepPavlov 进行文本分类聊天机器人框架。

问题是我没有足够的训练数据。理想情况下，我希望每个类只用几个样本进行文本分类。

Answer 1

你应该看看 DeepPavlov's autoFAQ models .这些模型是专门为在训练数据有限时有效而开发的。

可供您使用的模型很少

• 基于 tf-idf 的模型

• fastText 模型

• 以及两者的结合

更改配置文件中的数据集源，运行训练模型

python -m deeppavlov train tfidf_logreg_en_faq

您可以通过命令行与经过训练的模型进行交互

python -m deeppavlov interact tfidf_logreg_en_faq -d

或通过 Python 代码

from deeppavlov.core.commands.infer import build_model
faq = build_model("tfidf_logreg_en_faq", load_trained = True, download = True)
a = faq(["I need help"])
a

您可以在 colab notebook 中找到所有必需的代码片段