Facebook發佈對話研究框架ParlAI

編程語言 Facebook DIALOG Lua 量子位 2017-05-07

李林 編譯整理
量子位 出品 | 公眾號 QbitAI

Facebook發佈對話研究框架ParlAI

今天,Facebook發佈了開源的對話研究軟件框架ParlAI。

GitHub地址:facebookresearch/ParlAI

這個框架的目標是為研究者提供以下功能:

  • 一個訓練和測試對話模型的統一框架;

  • 同時在多個數據集上進行多任務訓練;

  • 與亞馬遜Mechanical Turk無縫整合,以便進行數據收集和人工評估。

該框架的第一版包含了20多種常用數據集:SQuAD、bAbI tasks、MCTest、WikiQA、WebQuestions、SimpleQuestions、WikiMovies、QACNN & QADailyMail、CBT、BookTest、bAbI Dialog tasks、Ubuntu Dialog、OpenSubtitles、Cornell Movie以及VQA-COCO2014。

另外,框架還包含PyTorch和Lua Torch下的訓練模型示例,既有GPU組訓練也有CPU訓練。

ParlAI主要維護者之一Jason Weston在Facebook上說,他們希望藉助社區力量來增加框架上訓練的任務和agent數量。該框架另外兩名主要維護者是Alexander H. Miller和Will Feng。

對Facebook人工智能實驗室現有的文本研究工具來說是一種補充,例如文本分類工具FastText、通用人工智能開發框架CommAI等。(完)


歡迎關注我們的公眾號:量子位(QbitAI),一起追蹤人工智能領域技術和產品動態。

相關推薦

推薦中...