2025Completed
Machine Learning Topic Classifier
A machine learning project that classifies news articles into predefined topics using a fine-tuned BERT deep learning model.
PythonFlaskBERTscikit-learnGoogle ColabpandasNumPyMatplotlib
1 / 2
Key Features
Automatic news article classification
Fine-tuned BERT model with 2,000+ labeled samples
Optimized training for best validation performance
Comparison with TF-IDF + Naive Bayes baseline
Flask backend for model inference
User-friendly web interface
Gdown for large model file handling
Challenges & Solutions
Fine-tuning BERT for optimal performance, TF-IDF and Naive Bayes baseline implementation.
Project Details
Year2025
StatusCompleted
CategoryMachine Learning Project