IMN

شبكة مطابقة تفاعلية لاختيار الاستجابة متعددة المنعطفات

يحتوي هذا المستودع على الكود المصدري ومجموعات البيانات لورقة CIKM 2019 لشبكة المطابقة التفاعلية لاختيار الاستجابة متعددة الأدوار في Chatbots المستندة إلى الاسترجاع بواسطة Gu et al.

حققت شبكة المطابقة التفاعلية (IMN) المقترحة لدينا أداءً جديدًا متطورًا في أربع مجموعات بيانات واسعة النطاق متاحة للجمهور للبحث في المحادثة متعددة المنعطفات.

نظرة عامة على النموذج

نتائج

التبعيات

بايثون 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 الذي يسجل النتائج لكل زوج من أزواج الاستجابة للسياق. قم بتشغيل الأمر التالي ويمكنك حساب مقياس الاستدعاء.

 python compute_recall.py

استشهد

إذا كنت تستخدم الكود ومجموعات البيانات، فيرجى الاستشهاد بالمقالة التالية: "شبكة المطابقة التفاعلية لاختيار الاستجابة متعددة الأدوار في روبوتات الدردشة القائمة على الاسترجاع" Jia-Chen Gu، Zhen-Hua Ling، Quan Liu. سيكم (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},
}