时间戳

首页 > 解决方案 > 如何使用 DeepPavlov 进行文本分类

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

问题描述

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

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

标签： nlpartificial-intelligencechatbot

解决方案

您应该查看DeepPavlov 的 autoFAQ 模型。当训练数据有限时，专门开发了一些有效的模型。

您可以使用的模型很少

  • 基于 tf-idf 的模型

  • 快速文本模型

  • 和两者的混合

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

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 笔记本中找到所有必需的代码片段

推荐阅读