oaarnikoivu / emotion-classifier

An attention-based BiLSTM for emotion classification.

An attention-based BiLSTM for emotion classification.

This model was trained to detect multiple emotions from text. The model achieves comparable results to the state-of-the-art by using an Attention BiLSTM with contextualized embeddings produced by a frozen BERT transformer model.

Overview

• /client contains all client-side code related to the UI of the application.
• /server contains the server-side code related to the application.
• /notebooks contains all the iPython notebooks to carry out the methods & experimental setup.

Technologies

The model architecture is constructed using PyTorch. The HuggingFace Transformers library is made use of in order to apply and retrive the embeddings from a pre-trained BERT model. Scikit Learn is also made use of to assess model performance using the Jaccard index, and micro-averaged and macro-averaged F1 scores, as well as for the implemenation of the baseline machine learning algorithms. The models were trained with Google Colaboratory notebooks using an NVIDIA Tesla K80 GPU.

Dataset

The dataset used is the SemEval Task 1: Affect in Tweets emotion classification dataset where given a tweet, the task is to classify the text as having no emotion or as one, or more, emotions for eight of the Plutchik categories plus optimism, pessimism, and love. The dataset consists of 6838 training examples, 886 validation examples and 3259 testing examples.

Report

emotion-classifier's Issues

Model availability

By Any chance is there a way to release the model to public or at least the data set info used for training?