Downloads · 30 days
51
5% of all-time downloads
bert-base-uncased_English_MultiLable_classification
Tobias/bert-base-uncased_English_MultiLable_classification
bert-base-uncased_English_MultiLable_classification is a text classification model from Tobias. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
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…
All-time downloads
943
Public
Repo size
438 MB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.h5438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- May 21, 2022
- Updated
- May 21, 2022
- SHA
- 18d2fd18
Try a prompt
Model card
From the Hugging Face model README
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
| Classes | Precision | Recall | F1 Score |
|---|---|---|---|
| Room | 77.78% | 77.78% | 77.78% |
| Location | 95.45% | 95.45% | 95.45% |
| Staff | 75.00% | 93.75% | 83.33% |
| Unknown | 71.43% | 50.00% | 58.82% |
| HotelOrganisation | 27.27% | 30.00% | 28.57% |
| Food | 87.50% | 87.50% | 87.50% |
| ReasonForStay | 63.64% | 58.33% | 60.87% |
| GeneralUtility | 66.67% | 50.00% | 66.67% |
| Accuracy | 74.00% | ||
| Macro Average | 70.59% | 67.85% | 68.68% |
| Weighted Average | 74.17% | 74.00% | 73.66% |
Confusion Matrix
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Languages
- eng
- Created
- May 21, 2022
- Updated
- May 21, 2022