Text Classification
Transformers
PyTorch
Safetensors
Russian
bert
russian
classification
sentiment
emotion-classification
multiclass
text-embeddings-inference
Instructions to use cointegrated/rubert-tiny2-cedr-emotion-detection with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use cointegrated/rubert-tiny2-cedr-emotion-detection with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="cointegrated/rubert-tiny2-cedr-emotion-detection")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("cointegrated/rubert-tiny2-cedr-emotion-detection") model = AutoModelForSequenceClassification.from_pretrained("cointegrated/rubert-tiny2-cedr-emotion-detection", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit ·
cd3543a
1
Parent(s): 074a2d0
Update README.md
Browse files
README.md
CHANGED
|
@@ -27,6 +27,6 @@ The quality of the predicted probabilities on the test dataset is the following:
|
|
| 27 |
|
| 28 |
| label | no emotion | joy |sadness |surprise| fear |anger | mean | mean (emotions) |
|
| 29 |
|----------|------------|--------|--------|--------|--------|--------| --------| ----------------|
|
| 30 |
-
| AUC | 0.
|
| 31 |
| F1 micro | 0.8624 | 0.9389 | 0.9362 | 0.9469 | 0.9575 | 0.9261 | 0.9280 | 0.9411 |
|
| 32 |
| F1 macro | 0.8562 | 0.8962 | 0.9017 | 0.8366 | 0.8359 | 0.6820 | 0.8348 | 0.8305 |
|
|
|
|
| 27 |
|
| 28 |
| label | no emotion | joy |sadness |surprise| fear |anger | mean | mean (emotions) |
|
| 29 |
|----------|------------|--------|--------|--------|--------|--------| --------| ----------------|
|
| 30 |
+
| AUC | 0.9286 | 0.9512 | 0.9564 | 0.8908 | 0.8955 | 0.7511 | 0.8956 | 0.8890 |
|
| 31 |
| F1 micro | 0.8624 | 0.9389 | 0.9362 | 0.9469 | 0.9575 | 0.9261 | 0.9280 | 0.9411 |
|
| 32 |
| F1 macro | 0.8562 | 0.8962 | 0.9017 | 0.8366 | 0.8359 | 0.6820 | 0.8348 | 0.8305 |
|