IMN

Интерактивная сеть согласования для многооборотного выбора ответа

Этот репозиторий содержит исходный код и наборы данных для документа CIKM 2019 «Интерактивная сеть сопоставления для многоходового выбора ответа в чат-ботах на основе поиска», автор 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

Цитировать

Если вы используете код и наборы данных, цитируйте следующую статью: «Интерактивная сеть сопоставления для многооборотного выбора ответов в чат-ботах на основе поиска» Цзя-Чен Гу, Чжэнь-Хуа Лин, Цюань Лю. ЦИКМ (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},
}