This repo contains the PyTorch implementation of the paper by Tianxiao Shen, Tao Lei, Regina Barzilay, and Tommi Jaakkola. NIPS 2017. arXiv
Instructions to use the code:
- Compile the protocol buffer in src/parameters.proto using the command:
protoc --proto_path=src --python_out=src src/parameters.proto
For furhter instructions on how to install protoc and what are protocol buffers, you can refere to Google's page
-
Update the parameters value contained in the file resources/params.asciipb with your choiche of the parameters
-
Train the model using the file in scripts/train_yelp.sh