IMN

เครือข่ายการจับคู่แบบโต้ตอบสำหรับการเลือกการตอบสนองแบบหลายเทิร์น

พื้นที่เก็บข้อมูลนี้ประกอบด้วยซอร์สโค้ดและชุดข้อมูลสำหรับ CIKM 2019 Paper Interactive Matching Network สำหรับการเลือกการตอบสนองแบบหลายเทิร์นใน Chatbots แบบดึงข้อมูลโดย Gu et al

Interactive Matching Network (IMN) ที่เรานำเสนอได้รับประสิทธิภาพการทำงานที่ล้ำสมัยชุดข้อมูลขนาดใหญ่สี่ชุดที่เปิดเผยต่อสาธารณะสำหรับการวิจัยเกี่ยวกับการสนทนาแบบหลายรอบ

ภาพรวมโมเดล

ผลลัพธ์

การพึ่งพาอาศัยกัน

หลาม 2.7
เทนเซอร์โฟลว์ 1.4.0

ชุดข้อมูล

คุณสามารถดาวน์โหลดชุดข้อมูลที่ประมวลผลแล้วซึ่งใช้ในรายงานของเราได้ที่นี่ และแตกไฟล์ลงในโฟลเดอร์ data
อูบุนตู_V1
อูบุนตู_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

อ้างอิง

หากคุณใช้โค้ดและชุดข้อมูล โปรดอ้างอิงเอกสารต่อไปนี้: "เครือข่ายการจับคู่เชิงโต้ตอบสำหรับการเลือกการตอบสนองแบบหลายเทิร์นในแชทบอทแบบดึงข้อมูล" 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},
}