Skip to content
master
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

GLKS

This is the implementation of GLKS proposed in "Thinking Globally, Acting Locally: Distantly Supervised Global-to-Local Knowledge Selection for Background Based Conversation".

1. Download data from "https://github.com/nikitacs16/Holl-E" and use "Prepare_holl.py" to prepare the dataset.

  • (1) Download the Holl-E dataset, make sure "holl\raw_data-20190221T150829Z-001\raw_data" contains the train, dev, test files.

  • (2) Change the configurations ("input_file", "version", "output_file") in "Prepare_holl.py" to prepare different versions of datasets.

  • (3) Download "glove.6B.300d" and put it in "holl/"

  • (4) Create a vocabulary file based on the training set of different versions in the format: "token \t token_frequency".

2. Change the configurations in "Run_GLKS.py" and run "python -m torch.distributed.launch --nproc_per_node=num_GPU Run_GLKS.py --mode='train/test'" to train or test models on different dataset versions.

Please cite our paper if you use the code:

@inproceedings{ren-2020-thinking,

Author = {Ren, Pengjie and Chen, Zhumin and Monz, Christof and Ma, Jun and de Rijke, Maarten},

Booktitle = {The 34th AAAI Conference on Artificial Intelligence},

Title = {Thinking Globally, Acting Locally: Distantly supervised global-to-local knowledge selection for background based conversation},

Year = {2020}}

About

No description, website, or topics provided.

Resources

Releases

No releases published

Packages

No packages published

Languages