Skip to content

UmerrAhsan/Text-Classification-using-Bert-

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Text-Classification-using-Bert-

Overview

In this project, I utilized PyTorch, the Transformers library, and Hugging Face's BERT model for fine-tuning on a text classification task. The training and evaluation loops were written in PyTorch, allowing efficient GPU utilization for model training. The primary goal was to achieve optimal performance on various metrics such as precision, recall, accuracy, and F1 score.

Dependencies

  • PyTorch
  • Transformers Library
  • Hugging Face BERT Model
  • GPU for accelerated fine-tuning

About

Text Classification using Bert

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published