/ Модель

Характеристики

Контекст
8 192 токена
Параметры
149 млн
Лицензия
MIT
Архитектура
modernbert · dense
Базовая модель
ModernBERT-base
Библиотека
transformers
Опубликована
14 янв 2025

modernbert-base-go-emotions

Классификация текстаТекст → label3 профессии

Дообученная ModernBERT-base классифицирует английские тексты по 28 эмоциональным меткам, допуская несколько меток для одного текста.

2 798скачиваний за 30 дней13лайковвеса обновлены13 сен 2026

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Модель cirimus/modernbert-base-go-emotions — дообученная на наборе GoEmotions англоязычная модель ModernBERT-base для многометочной классификации эмоций в тексте. Она присваивает каждому фрагменту текста один или несколько из 28 эмоциональных классов.

Модель подходит для анализа эмоциональных реакций в публикациях и комментариях, оценки отзывов клиентов и исследовательского анализа поведения. Она построена на архитектуре ModernBERT с 149 млн параметров и поддерживает контекст до 8192 токенов. Для обучения использовался корпус комментариев Reddit GoEmotions; модель выпущена в марте 2026 года.

На тестовой части GoEmotions при пороге 0,5 указаны macro F1 0,500 и micro F1 0,607; подбор отдельных порогов для меток может повысить полноту и macro F1. Качество может снижаться на текстах из других источников и культурных контекстов, особенно для редких классов; разметка корпуса также содержит неоднозначности. Лицензия — MIT.

Сценарии применения

  • определение эмоций в постах и комментариях на английском языке
  • анализ эмоциональной окраски отзывов клиентов
  • сегментация текстовых откликов по эмоциональным категориям
  • подготовка разметки эмоций для исследовательского анализа поведения

Ограничения

Корпус основан на комментариях Reddit, поэтому результаты могут хуже переноситься на другие жанры и культурные контексты; редкие классы и неоднозначная разметка ограничивают качество. Указанные метрики относятся к тестовой выборке GoEmotions.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

banner

ModernBERT Go Emotions 2026-03

Note: This version significantly improves the baseline 2025 model.

Overview

This model was fine-tuned from ModernBERT-base on the GoEmotions dataset for multi-label classification. It predicts emotional states in text, with a total of 28 possible labels. Each input text can have one or more associated labels, reflecting the multi-label nature of the task.

Try it out here.


Intended Use

The model is designed for emotion classification in English-language text, particularly in domains such as:

  • Social media sentiment analysis
  • Customer feedback evaluation
  • Behavioral or psychological research

Example Usage

Here’s how to use the model with Hugging Face Transformers:

from transformers import pipeline
import torch

# Load the model
classifier = pipeline(
    "text-classification", 
    model="cirimus/modernbert-base-go-emotions",
    top_k=5
)

text = "I am so happy and excited about this opportunity!"
predictions = classifier(text)

# Print top 5 detected emotions
print("\nTop 5 emotions detected:")
for pred in predictions[0]:  # Note: predictions is [[...]] format
    print(f"\t{pred['label']:10s} : {pred['score']:.3f}")

# Example output
# Top 5 emotions detected:
#        excitement : 0.976
#        joy        : 0.698
#        approval   : 0.012
#        optimism   : 0.007
#        admiration : 0.006

Model Details

  • Base Model: ModernBERT-base
  • Fine-Tuning Dataset: GoEmotions
  • Number of Labels: 28
  • Problem Type: Multi-label classification
  • Language: English
  • License: MIT
  • Fine-Tuning Framework: Hugging Face Transformers
  • Release: March 2026
  • Version: 2026-03 (Optimized)

Training Improvements

ParameterBaseline2026-03
Learning Rate2e-51.8e-4
LR ScheduleLinearCosine + Warmup
OptimizerAdamWAdafactor
Weight Decay0.010.001
Batch Size1632
New: EMANone0.999 decay
New: Dropout TuningDefaultClassifier: 0.1, Attention: 0.05, MLP: 0.05

Overall Results

SplitAccuracyPrecisionRecallMacro F1Micro F1
Train0.9760.7460.5080.5730.672
Validation0.9710.6220.4440.4980.603
Test0.9710.6520.4430.5500.607

Dataset

The GoEmotions dataset is a multi-label emotion classification dataset derived from Reddit comments. It contains 58,000 examples with 28 emotion labels (e.g., admiration, amusement, anger, etc.), and it is annotated for multi-label classification.


Evaluation Results

The model was evaluated on the test split of the GoEmotions dataset, using a threshold of 0.5 for binarizing predictions. The overall metrics were:

Standard Results:

Using the default threshold of 0.5.

LabelAccuracyPrecisionRecallF1MCCSupportThreshold
macro avg0.9710.6520.4430.5000.50654270.5
admiration0.9470.7190.7120.7160.6875040.5
amusement0.9850.8050.9050.8520.8462640.5
anger0.9700.6390.3840.4790.4811980.5
annoyance0.9410.5090.1720.2570.2723200.5
approval0.9430.6370.2910.3990.4063510.5
caring0.9770.5610.2740.3680.3821350.5
confusion0.9750.5940.3730.4580.4591530.5
curiosity0.9520.5380.5420.5400.5152840.5
desire0.9870.6040.3860.4710.476830.5
disappointment0.9740.6550.1260.2110.2801510.5
disapproval0.9530.5480.2550.3480.3532670.5
disgust0.9810.6180.3820.4720.4771230.5
embarrassment0.9950.7500.4050.5260.549370.5
excitement0.9840.6480.3400.4460.4621030.5
fear0.9920.7140.7050.7100.705780.5
gratitude0.9910.9520.9060.9290.9243520.5
grief0.9990.0000.0000.0000.00060.5
joy0.9800.7090.5590.6250.6191610.5
love0.9820.7760.8450.8090.8002380.5
nervousness0.9960.6430.3910.4860.500230.5
optimism0.9760.7210.5000.5900.5891860.5
pride0.9980.7500.3750.5000.529160.5
realization0.9740.5760.1310.2130.2661450.5
relief0.9981.0000.0910.1670.301110.5
remorse0.9920.5700.8040.6670.673560.5
sadness0.9790.6840.5000.5780.5751560.5
surprise0.9780.5960.4400.5060.5011410.5
neutral0.7980.7330.6090.6660.52817870.5

Optimal Results:

Using the best threshold for each label based on the training set (tuned on F1).

LabelAccuracyPrecisionRecallF1MCCSupportThreshold
macro avg0.9680.5830.5350.5500.5365427various
admiration0.9440.6800.7580.7170.6875040.39
amusement0.9840.8180.8520.8350.8262640.60
anger0.9650.5220.4700.4950.4771980.35
annoyance0.9310.4050.3720.3880.3513200.34
approval0.9260.4250.4250.4250.3853510.28
caring0.9730.4560.3480.3950.3851350.37
confusion0.9720.5000.4510.4740.4601530.39
curiosity0.9490.5110.6800.5830.5632840.40
desire0.9860.5710.4340.4930.491830.35
disappointment0.9660.3890.3840.3870.3691510.24
disapproval0.9360.3870.5020.4370.4082670.26
disgust0.9790.5500.4880.5170.5081230.32
embarrassment0.9950.7270.4320.5420.559370.31
excitement0.9800.4570.4170.4370.4271030.30
fear0.9910.6750.7180.6960.691780.45
gratitude0.9900.9520.8980.9240.9193520.56
grief0.9990.6670.3330.4440.47160.14
joy0.9780.6270.6150.6210.6091610.37
love0.9830.7810.8400.8100.8012380.54
nervousness0.9950.4550.4350.4440.442230.31
optimism0.9740.6300.5590.5930.5801860.32
pride0.9980.7500.3750.5000.529160.27
realization0.9680.3470.2340.2800.2691450.20
relief0.9980.6250.4550.5260.532110.20
remorse0.9920.5700.8040.6670.673560.50
sadness0.9770.6120.5450.5760.5651560.40
surprise0.9770.5680.4470.5000.4921410.46
neutral0.7910.6710.7180.6940.53717870.40

Confusion matrix:

banner


Limitations and Biases

  • Data Bias: The dataset is based on Reddit comments, which may not generalize well to other domains or cultural contexts.
  • Underrepresented Classes: Certain labels like "grief" and "relief" have very few examples, leading to lower performance for those classes.
  • Ambiguity: Some training data contain annotation inconsistencies or ambiguities that may impact predictions.


Environmental Impact

  • Hardware Used: NVIDIA RTX4090
  • Training Time: <1 hour
  • Carbon Emissions: ~0.04 kg CO2 (calculated via ML CO2 Impact Calculator).

Citation

If you use this model, please cite it as follows:

@inproceedings{JdFE2025b,
  title = {Emotion Detection with ModernBERT},
  author = {Enric Junqu\'e de Fortuny},
  year = {2025},
  howpublished = {\url{https://huggingface.co/cirimus/modernbert-base-go-emotions}},
}

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · transformers · text-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
ModernBertForSequenceClassification
Ревизия весов
598fc9b

0 моделей