Skip to content

rauni-iitr/Sentence-Classification-transformers-and-torch-training

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits

.gitignore

.gitignore

 
 

readme.md

readme.md

 
 

sent_classification_torch.ipynb

sent_classification_torch.ipynb

 
 

sent_classification_trainer.ipynb

sent_classification_trainer.ipynb

 
 

Repository files navigation

Text Classification - Sentiment Analysis

  • DistillBERT model fine tuned for sentence classification.

  • You can find two notebooks(documented):

    • with suffix "_trainer" - for transformers' Trainer API based training
    • with suffix "_torch" - for torch based training

About

Fine-Tuning transformer model(DistillBERT) for text classification using transformers Trainer API

Topics

transformers language-model fine-tuning

Resources

Readme
Activity

Stars

0 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages