Skip to content

Latest commit

 

History

History
31 lines (21 loc) · 1.41 KB

README.md

File metadata and controls

31 lines (21 loc) · 1.41 KB

Project: Emotion Detection from Text

###Team :

Highlights:

  • This is a multi-class text classification (sentence classification) problem.
  • The purpose of this project is to classify emotions from sentences into 7 different categories.
  • The model was built with Convolutional Neural Network (CNN) and Word Embeddings on Tensorflow.

Train:

  • To run CNN for classifcation -- Command: python train_cnn.py
  • To run LSTM for classification -- Command: python lstm.py

A directory will be created during training, and the best model will be saved in this directory.

Reference:

In order to run this download Word2Vec and Glove Vectors, since the file sizes are very large.