Svensk BERT-modell för sentimentanalys
Tolkning av labels när man använder APIn
"Label 0": negativt sentiment
"Label 1": positivt sentiment
Den ursprungliga modellen
Den ursprungliga modellen består av "bert-base-swedish-cased" från https://huggingface.co/KB/bert-base-swedish-cased
Fine tuning
Modellen har sedan tränats ScandiSent ett dataset bestående av 10000 recensioner från trustpilot som kodats som positiva eller negativa (5000 positiva och 5000 negativa).
https://github.com/timpal0l/ScandiSent
Vid validering på 20% av datasetet presterade modellen enligt följande:
precision recall f1-score support
Negative 0.92 0.99 0.96 943
Positive 0.99 0.91 0.95 915
accuracy 0.95 1858
macro avg 0.96 0.95 0.95 1858
weighted avg 0.96 0.95 0.95 1858
- Downloads last month
- 5