Code for "Unpaired Sentiment-to-Sentiment Translation: A Cycled Reinforcement Learning Approach" (ACL 2018)
Switch branches/tags
Nothing to show
Clone or download
Latest commit af44ebd Aug 23, 2018

README.md

Unpaired-Sentiment-Translation

Code for "Unpaired Sentiment-to-Sentiment Translation: A Cycled Reinforcement Learning Approach" [pdf]

Requirements

  • ubuntu 16.04
  • python 3.5
  • tensorflow 1.4.1
  • nltk 3.2.5

Data

The details of input data are shown in "train-original" and "test-original" (the processed Yelp data). If you want to run our code on your own dataset, you first need to process your dataset into the specific data structure.

Run

CUDA_VISIBLE_DEVICES=2 nohup bash run_train.sh > log_train.txt &
CUDA_VISIBLE_DEVICES=2 nohup bash run_test.sh > log_test.txt &

Cite

To use this code, please cite the following paper:

Jingjing Xu, Xu Sun, Qi Zeng, Xuancheng Ren, Xiaodong Zhang, Houfeng Wang, Wenjie Li. Unpaired Sentiment-to-Sentiment Translation: A Cycled Reinforcement Learning Approach. ACL 2018.

bibtext:

@inproceedings{unpaired-sentiment-translation,
  author    = {Jingjing Xu and Xu Sun and Qi Zeng and Xuancheng Ren and Xiaodong Zhang and Houfeng Wang and Wenjie Li},
  title     = {Unpaired Sentiment-to-Sentiment Translation: A Cycled Reinforcement Learning Approach},
  booktitle = {ACL},
  year      = {2018}
}