Project Overview

This project implements a fine-tuned BERT model for text classification and compares its performance to a traditional TF-IDF + Logistic Regression baseline.

Contents

1. BERT Fine-Tuning: Code for fine-tuning a pre-trained BERT model on the AG News dataset.
2. TF-IDF + Logistic Regression: A baseline model using traditional feature extraction and classification on the AG News Dataset.