Edit model card

German Hotel Review Sentiment Classification

A model trained on German Hotel Reviews from Switzerland. The base model is the bert-base-german-cased. 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 90.48% 82.61% 86.36%
Negative 70.59% 92.31% 80.00%
Neutral 28.57% 13.33% 18.18%
Accuracy 76.00%
Macro Average 63.21% 62.75% 61.52%
Weighted Average 73.43% 76.00% 73.65%

Confusion Matrix

Confusion Matrix

Downloads last month
19