ID-LSTM-pytorch
ID-LSTM-pytorch copied to clipboard
navid5792
A Pytorch implementation of the AAAI 2018 Paper "Learning Structured Representation for Text Classification via Reinforcement Learning"
trafficstars
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
navid5792