IMN

Interaktives Matching-Netzwerk für die Multi-Turn-Antwortauswahl

Dieses Repository enthält den Quellcode und die Datensätze für den CIKM-Artikel „Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots“ von Gu et al. aus dem Jahr 2019.

Unser vorgeschlagenes Interactive Matching Network (IMN) hat eine neue, hochmoderne Leistung bei vier großen Datensätzen erzielt, die für die Forschung zu Multi-Turn-Konversationen öffentlich verfügbar sind.

Modellübersicht

Ergebnisse

Abhängigkeiten

Python 2.7
Tensorflow 1.4.0

Datensätze

Sie können die in unserem Artikel verwendeten verarbeiteten Datensätze hier herunterladen und in den data entpacken.
Ubuntu_V1
Ubuntu_V2
Douban
E-Commerce

Trainieren Sie ein neues Modell

Nehmen Sie als Beispiel Ubuntu_V1.

 cd scripts
bash ubuntu_train.sh

Der Trainingsprozess wird in der Datei log_train_IMN_UbuntuV1.txt aufgezeichnet.

Testen Sie ein trainiertes Modell

 bash ubuntu_test.sh

Der Testvorgang wird in der Datei log_test_IMN_UbuntuV1.txt aufgezeichnet. Und Sie können eine ubuntu_test_out.txt -Datei erhalten, die die Ergebnisse für jedes Kontext-Antwort-Paar aufzeichnet. Führen Sie den folgenden Befehl aus und Sie können die Recall-Metrik berechnen.

 python compute_recall.py

Zitieren

Wenn Sie den Code und die Datensätze verwenden, zitieren Sie bitte das folgende Dokument: „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},
}