You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This project implements an emotion classification model using a pre-trained transformer model (DistilBERT) to classify emotions based on text inputs. The model is trained on a dataset containing various emotional statements.
4
+
5
+
## Table of Contents
6
+
7
+
-[Overview](#overview)
8
+
-[Dataset](#dataset)
9
+
-[Installation](#installation)
10
+
11
+
## Overview
12
+
13
+
The goal of this project is to classify emotions expressed in text using natural language processing (NLP) techniques. We leverage the Hugging Face Transformers library to fine-tune a pre-trained DistilBERT model on our dataset.
14
+
15
+
## Dataset
16
+
17
+
The dataset used for training the model should have the following structure:
18
+
19
+
| content | sentiment |
20
+
|--------------------------------|-----------|
21
+
| alonzo feels angry | anger |
22
+
| alonzo feels sad | sadness |
23
+
| alonzo feels terrified | fear |
24
+
25
+
Make sure to place your dataset in the project directory and name it `emotion_data.csv`.
26
+
27
+
## Installation
28
+
29
+
To run this project, you'll need to install the required Python packages. You can do this using pip:
0 commit comments