Skip to content

Commit d94d605

Browse files
Create README.md
1 parent 804344c commit d94d605

File tree

1 file changed

+32
-0
lines changed
  • Classification Models/Emotion Classification

1 file changed

+32
-0
lines changed
Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
# Emotion Classification from Text
2+
3+
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:
30+
31+
```bash
32+
pip install transformers torch pandas scikit-learn

0 commit comments

Comments
 (0)