ParlAI - AI 对话模型研究和训练框架


BSD
跨平台
Python

软件简介

ParlAI(发音为“par-lay”)是 Facebook 开源的,用于在 Python 中实现的对话 AI 研究框架。

其目标是为研究人员提供:

  • 一个用于训练和测试对话模型的统一框架

  • 一次对多个数据集进行多任务训练

  • 无缝整合 Amazon Mechanical Turk 进行数据收集和人工评估

目标

统一评估模型对话框架

  • 按需下载任务/数据集,并为其提供了同样简单的接口

  • 统一的数据集输入和评估框架/指标

  • agents/ 目录鼓励研究人员将培训代码提交到存储库以便与他人分享

  • 协助重现

最终目标是实现通用的对话,其中包含许多不同的技能

  • 无缝结合模拟和实际语言任务

  • 鼓励多任务模式开发和评估

  • 有助于减少模型到特定数据集的过度拟合

最终目标是实现与人的真实对话

  • 通过 Mechanical Turk 对人类的现场对话进行训练和评估

  • 只需简单的设置,就可以连接 Mechanical Turk 上的人类与你的对话代理

  • 允许比较不同研究组的 Turk 实验

一组数据集,以引导人工交互的工作对话模型

  • 激励构建将在此存储库中进行的新数据集