Tobias/bert-base-uncased_English_Hotel_classification

German Hotel Review Sentiment Classification

A model trained on English Hotel Reviews from Switzerland. The base model is the 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

Confusion Matrix

[image]