README.md

---
title: Custom Text Classification Studio
emoji: 🏷️
colorFrom: red
colorTo: blue
sdk: gradio
app_file: app.py
pinned: false
---

# Custom Text Classification Studio

An interactive educational machine learning application designed to help digital humanities and social science students learn supervised learning. Students can upload a labeled dataset, train a classifier locally, and instantly test predictions in real-time.

### Features
1. **Interactive Training**: Choose between Multinomial Naive Bayes, Logistic Regression, or Linear SVM models.
2. **Visual Diagnostic Dashboards**: View accuracy scores, comprehensive classification reports, and interactive Plotly confusion matrix heatmaps.
3. **Live Testing Playground**: Type or paste new, unseen paragraphs and view predicted categories along with a full probability distribution bar chart.
4. **Zero-Server dependencies**: Trains in seconds completely inside the Space memory.