Downloads · 30 days
0
arjahojnik/BiLSTM-sentiment-model
BiLSTM-sentiment-model is a machine learning model from arjahojnik. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This is a fine-tuned BiLSTM model trained for sentiment analysis on the TripAdvisor dataset. The model predicts sentiment scores on a scale of 1 to 5 based on review text.
Downloads · 30 days
0
Access
Public
Updated Jan 26, 2025
Repo size
359 MB
Likes
0
Public
Click a slice to open those files.
.txt347 MB · 97%
From the Hugging Face model README
This is a fine-tuned BiLSTM model trained for sentiment analysis on the TripAdvisor dataset. The model predicts sentiment scores on a scale of 1 to 5 based on review text.
nhull/tripadvisor-split-dataset-v2This model is designed to classify hotel reviews based on their sentiment. It assigns a star rating between 1 and 5 to a review, indicating the sentiment expressed in the review (1 = very bad, 2 = bad, 3 = neutral, 4 = good, 5 = very good).
The dataset used for training, validation, and testing is nhull/tripadvisor-split-dataset-v2. It consists of:
All splits are balanced across five sentiment labels.
| Metric | Value |
|---|---|
| Accuracy | 0.6167 |
| Precision | 0.62 |
| Recall | 0.62 |
| F1-Score | 0.62 |
| Label | Precision | Recall | F1-Score | Support |
|---|---|---|---|---|
| 1 | 0.73 | 0.71 | 0.72 | 1600 |
| 2 | 0.53 | 0.47 | 0.50 | 1600 |
| 3 | 0.52 | 0.60 | 0.56 | 1600 |
| 4 | 0.56 | 0.55 | 0.55 | 1600 |
| 5 | 0.74 | 0.75 | 0.74 | 1600 |
| True \ Predicted | 1 | 2 | 3 | 4 | 5 |
|---|---|---|---|---|---|
| 1 | 251 | 64 | 4 | 1 | 0 |
| 2 | 106 | 159 | 52 | 2 | 1 |
| 3 | 17 | 88 | 168 | 37 | 10 |
| 4 | 3 | 14 | 88 | 128 | 87 |
| 5 | 5 | 6 | 19 | 66 | 224 |
correct_predictions_BiLSTM.csv: Contains correctly classified reviews with their real and predicted labels.misclassified_predictions_BiLSTM.csv: Contains misclassified reviews with their real and predicted labels, along with the difference.glove.6B.100d.txt: Pre-trained 100-dimensional GloVe embeddings used for initializing the embedding layer in the models.