IMN
1.0.0
Este repositório contém o código-fonte e os conjuntos de dados do artigo CIKM 2019 Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots de Gu et al.
Nossa proposta Interactive Matching Network (IMN) alcançou um novo desempenho de última geração em quatro conjuntos de dados de grande escala que estão disponíveis publicamente para pesquisa em conversas multiturnas.
Pitão 2.7
Tensorflow 1.4.0
Você pode baixar os conjuntos de dados processados usados em nosso artigo aqui e descompactá-los na pasta de data .
Ubuntu_V1
Ubuntu_V2
Douban
Comércio eletrônico
Veja o Ubuntu_V1 como exemplo.
cd scripts
bash ubuntu_train.sh
O processo de treinamento é registrado no arquivo log_train_IMN_UbuntuV1.txt .
bash ubuntu_test.sh
O processo de teste é registrado no arquivo log_test_IMN_UbuntuV1.txt . E você pode obter um arquivo ubuntu_test_out.txt que registra pontuações para cada par de resposta de contexto. Execute o seguinte comando e você poderá calcular a métrica de Recall.
python compute_recall.py
Se você usar o código e os conjuntos de dados, cite o seguinte artigo: "Rede de correspondência interativa para seleção de resposta multivoltas em chatbots baseados em recuperação" 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},
}
