Skip to content

This AI project aims to classify people's reviews from watching films using the IMDb dataset and an encoder-based transformer architecture, specifically the well-known LLM model BERT.

Notifications You must be signed in to change notification settings

khaireddine-satouri/Text-classification-using-Transformers

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fine tuning BERT model for text classification

This project demonstrates a text classification system using the DistilBERT model from HuggingFace's Transformers library. The model is trained on the IMDb dataset to classify text reviews as positive or negative.

About

This AI project aims to classify people's reviews from watching films using the IMDb dataset and an encoder-based transformer architecture, specifically the well-known LLM model BERT.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages