Модель распознавания эмоций в речи GigaAM-Emo
умеет классифицировать 4 эмоции: angry
, sad
, neutral
, positive
. Для некоторых задач бывает не достаточно классифицировать эмоции на 4 класса и хочется иметь более полный список эмоций. В статье рассмотрим: существующие корпуса данных для распознавания эмоций, ключевые возможности разработанного пайплайна для дообучения GigaAM
с использованием библиотек hydra
и Pytorch Lightning
, результаты экспериментов с различными архитектурами и функциями потерь. Демо можно попробовать тут
Обучаем GigaAM-Emo распознавать ещё больше эмоций