Skip to content

Jasmine090/AI-final-project

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

38 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI-final-project

Topic: Text-based emoji recommendation

Overview of the task:

Train an agent to find the best emoji choise for a sentence by using multi-class NLP model and Bert model.

Dataset:

Emotions dataset for NLP in Kaggle

Prerequisite:

Environment:
Google colab (baseline metrics)
Kaggle notebook with GPU (main model)

Packages version:
Python 3.6.6
numpy 1.18.2
pandas 0.25.3
transformers 3.0.2

Usage:
batch size = 32
#epochs = 3

Hyperparameters:
optimization algorithm: Adam learning rate optimization algorithm (type of Stochastic Gradient Descent with momentum; Momentum is described as the moving average of the gradient instead of gradient itself.)

Experiment results: Accuracy
image

About

Text-based emoji recommendation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Jupyter Notebook 100.0%