IMN

Jaringan Pencocokan Interaktif untuk Pemilihan Respons Multi-Putaran

Repositori ini berisi kode sumber dan kumpulan data untuk makalah CIKM 2019 Jaringan Pencocokan Interaktif untuk Seleksi Respons Multi-Giliran dalam Chatbots Berbasis Pengambilan oleh Gu dkk.

Jaringan Pencocokan Interaktif (IMN) yang kami usulkan telah mencapai kinerja mutakhir pada empat kumpulan data berskala besar yang tersedia untuk umum untuk penelitian percakapan multi-turn.

Ikhtisar model

Hasil

Ketergantungan

ular piton 2.7
Aliran Tensor 1.4.0

Kumpulan data

Anda dapat mengunduh kumpulan data olahan yang digunakan dalam makalah kami di sini dan mengekstraknya ke folder data .
Ubuntu_V1
Ubuntu_V2
Douban
E-niaga

Latih model baru

Ambil Ubuntu_V1 sebagai contoh.

 cd scripts
bash ubuntu_train.sh

Proses pelatihan dicatat dalam file log_train_IMN_UbuntuV1.txt .

Uji model terlatih

 bash ubuntu_test.sh

Proses pengujian dicatat dalam file log_test_IMN_UbuntuV1.txt . Dan Anda bisa mendapatkan file ubuntu_test_out.txt yang mencatat skor untuk setiap pasangan respons konteks. Jalankan perintah berikut dan Anda dapat menghitung metrik Recall.

 python compute_recall.py

Mengutip

Jika Anda menggunakan kode dan kumpulan data, harap kutip makalah berikut: "Jaringan Pencocokan Interaktif untuk Pemilihan Respons Multi-Putaran di Chatbot Berbasis Pengambilan" 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},
}