Convai2 submission.
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
docs Added NeurIPS 2018 slides. Dec 18, 2018
environment file path fix Oct 1, 2018
model Added license headers. (#6) Dec 25, 2018
.dockerignore Added ignore files. Oct 18, 2018
.gitignore Added ignore files. Oct 18, 2018
LICENSE Update LICENSE Dec 10, 2018
README.md Added NeurIPS 2018 slides. Dec 18, 2018
agent.py Added flag episode_done to reply. Oct 19, 2018
config.py fix max_seq_len Oct 1, 2018
eval_f1.py fix max_seq_len Oct 1, 2018
eval_hits.py fix max_seq_len Oct 1, 2018
interactive.py okey... Oct 4, 2018
requirements.txt Added test Dockerfile. Sep 30, 2018
train.py move parameters Sep 30, 2018
wild.py Used convai_word from ParlAI. Oct 18, 2018

README.md

ConvAI2

Team: Lost in Conversation /2

NeurIPS 2018 presentation.

Links

BPE vocabulary: https://www.dropbox.com/s/n2jbjyq32x6jgr6/parameters.zip?dl=1

Model checkpoint file: https://www.dropbox.com/s/cs6zd9yntn6ixea/last_checkpoint?dl=1

Team

How to run

Unzip BPE vocabulary files into ./parametes folder and save checkpoint into ./checkpoints folder or use scripts (see below).

The easiest way to prepare environment is to run script prepare_environment.sh. After that docker container with retrieval server must be run in demon mode and image with transformer_chatbot must be built. Run scripts from the root folder of this repository.

Retrieval server can be run with script run_retrieval_servet.sh. Server do not need the internet connection for its work, for connection with transformer_chatbot port 9200 is used (containers must be in the same docker network).

After preparations metrics can be evaluated with corresponding docker_*.sh scripts or *.py scripts can be used during interactive container run.

Usage of docker container for transformer_chatbot is not necessary, but retrieval server must always be run for correct work of transformer_chatbot.

List of used python modules is in requirements.txt. Also pytorch=0.4.1 is used.

Metrics

f1: 0.178

hits@1: 0.173