Dataset domain | Total | +ve | ‒ve |
---|---|---|---|
Restaurant from Yelp | 1,000 | 500 | 500 |
Mobile from Amazon | 1,000 | 500 | 500 |
Movies from IMDB | 1,000 | 500 | 500 |
Model | Single-layer Bi-LSTM ( | Two-layer Bi-LSTM | Two-layer HCT Bi-LSTM | |||
---|---|---|---|---|---|---|
Dataset | T | V | T | V | T | V |
Amazon | 0.83 | 0.51 | 0.91 | 0.70 | 0.95 | 0.76 |
Yelp | 0.84 | 0.70 | 0.85 | 0.72 | 0.86 | 0.75 |
IMDB | 0.71 | 0.81 | 0.90 | 0.81 | 0.95 | 0.82 |
Parameter | Value |
---|---|
Vocabulary size | 10,000 |
Bi-LSTM | 2 layer |
Dense | 1 |
Activation | Sigmoid |
Optimizer | Adam function |
Loss Function | Binary cross-entropy |
Input Length | 100 |
Learning rate | 0.002 |
Epoch | 10 |