Skip to content

minfenli/GoEmotions

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GoEmotions

A fineturned BERT on GoEmotions with Huggingface Transformers.

Description

  1. Based on the uncased BERT pretrained model with a linear output layer.
  2. Added several commonly-used emoji and tokens to the special token list of the tokenizer.
  3. Used weighted loss and focal loss with label smoothing while training.

How to Download Models

Make a directory for saving models.

$ mkdir model
$ cd model

Download models from Huggingface.

$ git clone https://huggingface.co/justin871030/bert-base-uncased-goemotions-original-finetuned
$ git clone https://huggingface.co/justin871030/bert-base-uncased-goemotions-ekman-finetuned
$ git clone https://huggingface.co/justin871030/bert-base-uncased-goemotions-group-finetuned

Training Parameters

Pretrained Model: BERT-Base, Uncased (12-layer, 768-hidden, 12-heads)

Parameter
Learning rate 5e-6
Warmup proportion 0.1
Epochs (Original) 20
Epochs (Else) 10
Max Seq Length 50
Batch size 16

Results

Best Results of Macro F1

Macro F1 (%) Dev Test
original 51.75 53.11
group 70.09 70.06
ekman 64.12 63.76

Reference

Releases

No releases published

Packages

No packages published