Downloads · 30 days
13
2% of all-time downloads
danish-bert-review-sentiment
KennethTM/danish-bert-review-sentiment
danish-bert-review-sentiment is a text classification model from KennethTM. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as cc-by-4.0.
BERT classification model for short customer reviews written in Danish.
All-time downloads
623
Public
Parameters
111M
885 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin443 MB · 50%
Parameter types
How the weights are stored.
F32111M · 100%
Try a prompt
Model card
From the Hugging Face model README
What is this?
BERT classification model for short customer reviews written in Danish.
The model uses 5 classes ranging from 1-5 stars:
- ⭐ (very poor)
- ⭐⭐ (poor)
- ⭐⭐⭐ (neutral)
- ⭐⭐⭐⭐ (good)
- ⭐⭐⭐⭐⭐ (very good)
The model is fine-tuned using the pre-trained Danish BERT model.
How to use
Test the model using the 🤗Transformers library pipeline:
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="KennethTM/danish-bert-review-sentiment")
classifier("Intet virkede og ingen hjælp at hente.")
#[{'label': '⭐', 'score': 0.4953940808773041}]
Or load it using the Auto* classes:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("KennethTM/danish-bert-review-sentiment")
tokenizer = AutoTokenizer.from_pretrained("KennethTM/danish-bert-review-sentiment")
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- cc-by-4.0
- Languages
- da
- Created
- Jun 8, 2023
- Updated
- Jul 5, 2023