Downloads · 30 days
8
0% of all-time downloads
sunny3/rubert-conversational-sentiment-balanced
rubert-conversational-sentiment-balanced is a text classification model from sunny3. Use it when you need a label for a piece of text. It is set up for transformers.
This model was initialized with rubert-base-cased-conversational weights and trained on a batch of datasets collected by Smetanin, using the same training sampling presented in this wonderful work. This approach allow…
Downloads · 30 days
8
0% of all-time downloads
All-time downloads
3.4K
Public
Parameters
178M
1.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.bin711 MB · 50%
How the weights are stored.
F32178M · 100%
From the Hugging Face model README
This model was initialized with rubert-base-cased-conversational weights and trained on a batch of datasets collected by Smetanin, using the same training sampling presented in this wonderful work. This approach allows for a uniform distribution among different datasets and three classes of sentiment labels: negative, neutral, and positive. Datasets were prepared by David Dale and are hosted here.
I chose rubert-base-cased-conversational weights because, according to Smetanin's work, this model ranks first among all other multilingual and popular Russian language models with BERT base architecture.
This model was trained and tested using the code and hyperparameters from the rubert-tiny-sentiment-balanced work.
There are only three labels: negative - 0, neutral - 1, positive - 2
It outperforms rubert-tiny-sentiment-balanced on four datasets, underperforms on one (linis), and has the same performance on mokoron and rureviews. See this for the comparison.
| Source | Macro F1 |
|---|---|
| SentiRuEval2016_banks | 0.88 |
| SentiRuEval2016_tele | 0.79 |
| kaggle_news | 0.73 |
| linis | 0.46 |
| mokoron | 0.98 |
| rureviews | 0.77 |
| rusentiment | 0.74 |