Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.
В подключённых источниках результатов для этой модели пока нет.
Что умеет модель
Описание составлено по карточке модели и метаданным Hugging Face
heegyu/gpt2-emotion — англоязычная модель генерации текста на базе GPT-2, разработанная Heegyu Kim и представленная в формате PyTorch для библиотеки Transformers. Она продолжает текст по затравке, начинающейся с одного из маркеров эмоций: sadness, joy, love, anger, fear или surprise; пример в карточке показывает генерацию текста после маркера sadness. Модель подходит для создания коротких эмоционально окрашенных продолжений и прототипов диалоговых или творческих приложений.
Контекстное окно составляет 1024 токена. Модель распространяется по лицензии MIT. Размер модели и конкретная конфигурация обучения в доступных данных не указаны.
Сценарии применения
генерация текста с заданным эмоциональным тоном
создание эмоционально окрашенных продолжений фраз
прототипирование генератора текстов по маркерам эмоций
Ограничения
Контекст ограничен 1024 токенами; генерация зависит от одного из шести заданных эмоциональных маркеров. Карточка не приводит количественной оценки качества.
Для каких профессий подходит3
Разделы каталога профессий, в задачах которых используется эта модель
from transformers import pipeline
generator = pipeline('text-generation', 'heegyu/gpt2-emotion')
prompt = "sadness I'm so " # start token should be one of ["sadness", "joy", "love", "anger", "fear", "surprise"]
print(generator(prompt)[0]['generated_text'])
>>> sadness I'm so tired of seeing all the stupid things that i ve learned from past years that i feel like ive been so stupid and blah and then i feel like ive just wasted all my energy doing stupid shit like how i never
Как запустить
На своей GPU в ailist. Цена — по текущим предложениям провайдеров