File size: 1,149 Bytes
a8714f4 610a19b a8714f4 610a19b a8714f4 610a19b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
language: eng
tags:
- bert
license: apache-2.0
widget:
- text: "The hotel is very nicely located"
example_title: "Example 1"
- text: "The reception staff were extremely helpful and very welcoming"
example_title: "Example 2"
- text: "There is no balcony in the rooms on the mountain side"
example_title: "Example 3"
- text: "A bit pricey"
example_title: "Example 4"
---
# German Hotel Review Sentiment Classification
A model trained on English Hotel Reviews from Switzerland. The base model is the [bert-base-uncased](https://huggingface.co/bert-base-uncased). The last hidden layer of the base model was extracted and a classification layer was added. The entire model was then trained for 5 epochs on our dataset.
# Model Performance
| Classes | Precision | Recall | F1 Score |
| :---: | :---: | :---: |:---: |
| Positive | 86.36% | 95.00% | 90.48% |
| Negative | 90.91% | 76.92% | 83.33% |
| Neutral | 50.00% | 42.86% | 46.15% |
| Accuracy | | | 83.00% |
| Macro Average | 75.76% | 71.59% | 73.32% |
| Weighted Average | 82.45% | 83.00% | 82.41% |
## Confusion Matrix
![Confusion Matrix](bert-base-uncased_English_sentiment.jpg) |