/Structure_Representation

A Pytorch implementation of the AAAI 2018 Paper "Learning Structured Representation for Text Classification via Reinforcement Learning"

Primary LanguagePython

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