--- license: apache-2.0 datasets: - seara/ru_go_emotions language: - ru base_model: - cointegrated/rubert-tiny2 pipeline_tag: text-classification --- # RuBert-tiny2-EmotionsDetected # RuBert-tyny2-EmotionsDetected This model was obtained by fine-tuning the model [RuBert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2) on the dataset [ru-goemotions](https://github.com/searayeah/ru-goemotions) containing 28 emotions: ``` 0: admiration (восхищение) 1: amusement (веселье) 2: anger (злость) 3: annoyance (раздражение) 4: approval (одобрение) 5: caring (забота) 6: confusion (непонимание) 7: curiosity (любопытство) 8: desire (желание) 9: disappointment (разочарование) 10: disapproval (неодобрение) 11: disgust (отвращение) 12: embarrassment (смущение) 13: excitement (возбуждение) 14: fear (страх) 15: gratitude (признательность) 16: grief (горе) 17: joy (радость) 18: love (любовь) 19: nervousness (нервозность) 20: optimism (оптимизм) 21: pride (гордость) 22: realization (осознание) 23: relief (облегчение) 24: remorse (раскаяние) 25: sadness (грусть) 26: surprise (удивление) 27: neutral (нейтральность) ``` At the moment, the model has been trained for 40 epochs with the following hyperparameters: ``` per_device_train_batch_size=16, per_device_eval_batch_size=16, num_train_epochs=1, weight_decay=0.01, learning_rate=1e-5, save_total_limit=2, load_best_model_at_end=True, metric_for_best_model="f1", greater_is_better=True ```