Fine-Tuning a Transformer Model with Hugging Face Transformers Library

Description

In this project, I fine-tuned a pre-trained Transformer model using Hugging Face's transformers library to perform sentiment analysis on IMDb reviews. The process involved tokenizing data with DistilBERT, training the model, evaluating its performance, and deploying it for sentiment classification tasks.

Utilities Used

• Hugging Face Transformers
• Hugging Face Datasets
• PyTorch
• Trainer API
• Google Colab

Project Walk-through

Dataset Loading, Tokenization, and Model Selection


Model Fine-Tuning and Evaluation


Model Deployment