mrm8488/t5-base-finetuned-emotion

Google's T5 base fine-tuned on emotion recognition dataset for Emotion Recognition downstream task . The training script is a slightly modified version of this Colab Notebook created by Suraj Patil . It allows to classifiy the text into one of the following 6 emotions: 'sadness,' 'joy', 'anger', 'surprise', 'love', 'fear' and 'disgusting' The model was presented in Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer by Colin Raffel, Noam Shazeer, Adam Roberts, Katherine Lee, Michael Matena, Yanqi Zhou, Wei Li, Peter J,