Text-Classification via RL Pytorch

Description

This repository contains the Pytorch implmentation of AAAI 2018 Paper "Learning Structured Representation for Text Classification via Reinforcement Learning".

Requirements

• Python==3.6.6
• PyTorch==0.4.0
• torchtext==0.2.3

Downloads and Setup

Once you clone this repo, run the main.py file to process the dataset and to train the model.

$ python main.py

References

• Learning Structured Representation for Text Classification via Reinforcement Learning : Paper