README.md

metadata

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.