Emotion Detection in Written Sentences

[nina-bu]

Student

• Nina Bu RA60/2020, Grupa 2

Asistent

• Branislav Anđelić

Problem koji se rešava

• Cilj ovog projekta je prepoznavanje emocije u rečenici, odnosno klasifikacija rečenica na osnovu emocija koje iskazuju.
• Za potrebe projekta, svakoj rečenici odgovara tačno jedna od šest navedenih emocija na osnovu kojih će se vršiti klasifikacija: sadness, anger, love, surprise, fear, joy

Algoritmi

• Rešenje problema je podeljeno u nekoliko koraka:
  1. Exploratory Data Analysis (EDA) - analiziranje podataka i sređivanje dataseta ako bude potrebno
  2. Pretprocesiranje i izdvajanje karakteristika - uklanjanje stop reči, lematizacija, primena TF-IDF itd.
  3. Obučavanje modela
  4. Naive Bayes
  5. Random Forest
  6. RNN (LSTM)

Podaci koji se koriste

• Koristiće se dataset: https://www.kaggle.com/datasets/praveengovi/emotions-dataset-for-nlp

Metrika za merenje performasni

• Performanse modela će se meriti korišćenjem accuracy metrike, tj. procentom ispravno klasifikovanih rečenica.

Validacija rešenja

• Dataset će biti podeljen na tri skupa: trening, validacioni i test skup u odnosu 70/20/10

[bandjeo]

Tema odobrena. Ostavite link ka github repozitorijumu.

Srećan ran!

[nina-bu]

Link do repozitorijuma: https://github.com/nina-bu/nlp-emotion-classifier