/Topic_Classification_SLM

This repository contains a PyTorch-based text classification model for the AG News dataset. The model uses embeddings and a fully connected layer for classification. Tokenization is done via NLTK, and the model is optimized using Adam. After 10 epochs, the model achieved a test accuracy of 91.45%.

Primary LanguagePythonMIT LicenseMIT

This repository is not active