IMN

다중 회전 응답 선택을 위한 대화형 매칭 네트워크

이 저장소에는 Gu 외의 검색 기반 챗봇에서 다중 회전 응답 선택을 위한 CIKM 2019 논문 Interactive Matching Network의 소스 코드와 데이터 세트가 포함되어 있습니다.

우리가 제안한 IMN(Interactive Matching Network)은 다중 회전 대화 연구를 위해 공개적으로 사용할 수 있는 4개의 대규모 데이터 세트에서 새로운 최첨단 성능을 달성했습니다.

모델 개요

결과

종속성

파이썬 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 논문을 인용해 주세요. 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},
}