/ Модель

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

Контекст
512 токенов
Размер весов
336 МБ
Лицензия
MIT
Архитектура
distilbert · dense
Опубликована
9 сен 2023

distilbert-base-uncased-go-emotions-onnx

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

Квантованная ONNX-модель DistilBERT определяет эмоции в тексте по категориям GoEmotions.

3 659скачиваний за 30 дней9лайковвеса обновлены9 сен 2023

Бенчмарки

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

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

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

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

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

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

Cohee/distilbert-base-uncased-go-emotions-onnx — модель классификации текста на основе DistilBERT, представленная в формате ONNX и квантованная для более эффективного выполнения. Она получена из joeddav/distilbert-base-uncased-go-emotions-student: модель-дистиллят обучали на неразмеченных примерах GoEmotions с псевдометками, созданными NLI-моделью для классификации без примеров. Модель предназначена для определения эмоциональных категорий в тексте и демонстрации более компактной альтернативы ресурсоёмкой zero-shot классификации.

Обучение проводилось 10 эпох со смешанной точностью. Контекст составляет до 512 токенов; архитектура — DistilBERT для классификации последовательностей. Исходный набор GoEmotions допускает несколько меток на пример, однако учитель назначал только одну метку, поэтому модель может хуже работать, чем вариант, обученный на полностью размеченных данных. Лицензия — MIT.

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

  • классификация эмоций в сообщениях и комментариях
  • анализ эмоционального тона коротких текстов
  • быстрая демонстрация классификатора эмоций без запуска NLI-модели
  • автоматическая предварительная разметка эмоциональных категорий

Ограничения

По README, качество может быть ниже, чем у модели, обученной с полной супервизией; псевдоразметка учителя была одно-меточной, хотя GoEmotions допускает несколько эмоций на пример.

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

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

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

Также подходит для

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

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

Открыть на Hugging Face

joeddav/distilbert-base-uncased-go-emotions-student converted to ONNX and quantized using optimum.


distilbert-base-uncased-go-emotions-student

Model Description

This model is distilled from the zero-shot classification pipeline on the unlabeled GoEmotions dataset using this script. It was trained with mixed precision for 10 epochs and otherwise used the default script arguments.

Intended Usage

The model can be used like any other model trained on GoEmotions, but will likely not perform as well as a model trained with full supervision. It is primarily intended as a demo of how an expensive NLI-based zero-shot model can be distilled to a more efficient student, allowing a classifier to be trained with only unlabeled data. Note that although the GoEmotions dataset allow multiple labels per instance, the teacher used single-label classification to create psuedo-labels.

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

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

GPU в ailist

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

Нельзя запустить

В репозитории нет весов в формате PyTorch/safetensors.

Класс архитектуры
DistilBertForSequenceClassification
Ревизия весов
d22488b

0 моделей