用於多輪響應選擇的互動式匹配網絡

該儲存庫包含 Gu 等人發表的 CIKM 2019 論文《Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots》的源代碼和資料集。

我們提出的互動式匹配網路（IMN）在四個可公開用於多輪對話研究的大型資料集上取得了最先進的效能。

型號概覽

結果

依賴關係

Python 2.7
張量流1.4.0

數據集

您可以在此處下載我們論文中使用的處理後的資料集並將其解壓縮到data資料夾中。
Ubuntu_V1
Ubuntu_V2
豆瓣
電子商務

訓練新模型

以Ubuntu_V1為例。

 cd scripts
bash ubuntu_train.sh

訓練過程記錄在log_train_IMN_UbuntuV1.txt檔案中。

測試經過訓練的模型

 bash ubuntu_test.sh

測試過程記錄在log_test_IMN_UbuntuV1.txt檔案中。您可以獲得一個ubuntu_test_out.txt文件，該文件記錄每個上下文回應對的分數。執行以下命令，您可以計算 Recall 的指標。

 python compute_recall.py

引用

如果您使用程式碼和資料集，請引用以下論文： “Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots” Jia-Chen Gu、Zhen-Hua Ling、Quan Liu。 CIKM (2019)

 @inproceedings{Gu:2019:IMN:3357384.3358140,
 author = {Gu, Jia-Chen and 
           Ling, Zhen-Hua and 
           Liu, Quan},
 title = {Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots},
 booktitle = {Proceedings of the 28th ACM International Conference on Information and Knowledge Management},
 series = {CIKM '19},
 year = {2019},
 isbn = {978-1-4503-6976-3},
 location = {Beijing, China},
 pages = {2321--2324},
 url = {http://doi.acm.org/10.1145/3357384.3358140},
 doi = {10.1145/3357384.3358140},
 acmid = {3358140},
 publisher = {ACM},
}

展開

附加信息