4 个月前

对话数据集仓库

对话数据集仓库

摘要

机器学习的进步往往受到大规模数据集的可用性和用于比较建模方法的一致评估指标的推动。为此,我们介绍了一个包含数亿个对话示例的数据集仓库,并提出了一种使用“100选1准确率”(1-of-100 accuracy)的标准评估程序,用于对话响应选择模型。该仓库中包含了脚本,这些脚本允许研究人员重现标准数据集,或者根据需要调整预处理和数据过滤步骤。我们引入并评估了几个具有竞争力的基线模型,这些模型的实现已共享在仓库中,并且还介绍了一种神经编码器模型,该模型是在整个训练集上进行训练的。

代码仓库

SarthakVaswani/ace_bot
GitHub 中提及
PolyAI-LDN/conversational-datasets
官方
tf
GitHub 中提及

基准测试

基准方法指标
conversational-response-selection-on-polyaiPolyAI Encoder
1-of-100 Accuracy: 61.3%
conversational-response-selection-on-polyai-1PolyAI Encoder
1-of-100 Accuracy: 30.6%
conversational-response-selection-on-polyai-2PolyAI Encoder
1-of-100 Accuracy: 71.3%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
对话数据集仓库 | 论文 | HyperAI超神经