Skip to content

A text classifier fine tuned on pre-trained BERT for Sarcasm Detection in News Headlines (PyTorch Implementation)

Notifications You must be signed in to change notification settings

mabdullah1994/Text-Classification-with-BERT-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text-Classification-with-BERT-PyTorch-Implementation

A text classifier by fine tuning on pre-trained BERT for Sarcasm Detection in News Headlines (PyTorch Implementation)

Kaggle Notebook

Transfer Learning for Text Data In Pytorch (BERT)

Dataset

News Headlines Dataset For Sarcasm Detection

Pre Requisite

  • python 3.7
  • torch 1.1.0
  • pytorch-transformers
  • pandas
  • numpy
  • tqdm

Results (Train Accuracy / Validation Accuracy)

  • Epoch 1 (69.89 / 81.16)
  • Epoch 2 (83.52 / 83.64)
  • Epoch 3 (87.21 / 82.70)
  • Epoch 4 (89.38 / 85.23)
  • Epoch 5 (91.09 / 85.54)
  • Epoch 6 (92.23 / 85.86)

About

A text classifier fine tuned on pre-trained BERT for Sarcasm Detection in News Headlines (PyTorch Implementation)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages