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
20
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.