install requirements
python >= 3.6.3
$ pip install requirements.txtdataset
- used the 100k subset of the Yelp Dataset
training and validating models
AWD-LSTM
$ python main.py --total_epochs 20SimpleLSTM
$ python main.py --model simple-lstmBidirectionalLSTM
$ python main.py --model bidirectional-lstmresults so far
| Model | Training Accuracy | Validation Accuracy |
|---|---|---|
| SimpleLSTM | 70.88% | 67.2% |
| BidirectionalLSTM | 70.75% | 68.56% |
| AWD-LSTM (no pretraining) | 72.85% | 69.37% |