Skip to content

Distilbert trained on goemotions dataset using huggingface

Notifications You must be signed in to change notification settings

darshpanchal/goemotions-distilbert

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

DistilBERT trained on GoEmotions dataset (Text Classification).

Disclaimer - GoEmotions is a multilabel and multiclass dataset, but I have made it multiclass by keeping just the first label in multilabel rows.

Files:

  • goemotions-distilbert.ipynb contains all training and basic inference code.

Future:

  • Convert to TFLite for mobile devices.
  • Deploying model.

References:

About

Distilbert trained on goemotions dataset using huggingface

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published