CoolFace
Modelpublic

Tobias/bert-base-uncased_German_MultiLable_classification

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
1likes48downloads
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

ClassesPrecisionRecallF1 Score
Room84.62%88.00%86.27%
Food79.17%82.61%80.85%
Staff63.64%70.00%66.67%
Location83.33%62.50%71.43%
GeneralUtilitys76.92%76.92%76.92%
HotelOrganisation26.67%30.77%28.57%
Unknown25.00%16.67%20.00%
ReasonForStay100.00%50.00%66.67%
Accuracy69.00%
Macro Average67.42%59.68%62.17%
Weighted Average69.36%69.00%68.79%

Confusion Matrix

[image]