IMN
1.0.0
Este repositorio contiene el código fuente y los conjuntos de datos para el artículo de CIKM 2019 Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots de Gu et al.
Nuestra propuesta Interactive Matching Network (IMN) ha logrado un nuevo rendimiento de vanguardia en cuatro conjuntos de datos a gran escala que están disponibles públicamente para la investigación sobre conversaciones de múltiples turnos.
Pitón 2.7
Tensorflow 1.4.0
Puede descargar los conjuntos de datos procesados utilizados en nuestro artículo aquí y descomprimirlos en la carpeta de data .
Ubuntu_V1
Ubuntu_V2
Douban
Comercio electrónico
Tomemos Ubuntu_V1 como ejemplo.
cd scripts
bash ubuntu_train.sh
El proceso de formación se registra en el archivo log_train_IMN_UbuntuV1.txt .
bash ubuntu_test.sh
El proceso de prueba se registra en el archivo log_test_IMN_UbuntuV1.txt . Y puede obtener un archivo ubuntu_test_out.txt que registra puntuaciones para cada par de contexto-respuesta. Ejecute el siguiente comando y podrá calcular la métrica de recuperación.
python compute_recall.py
Si utiliza el código y los conjuntos de datos, cite el siguiente artículo: "Red de coincidencia interactiva para la selección de respuestas de múltiples turnos en chatbots basados en recuperación" 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},
}
