Tobias/bert-base-uncased_English_MultiLable_classification

English Hotel Review MultiLabel 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]