spam-classifier-gradio

Senior Project Notice

This repository was created for a senior project in ENGT 375 Applied Machine Learning at Old Dominion University. It is provided for educational and research demonstration purposes only. It is not intended for production use, security filtering, or making real-world spam/phishing decisions. Always use established security tools for operational email protection.

Spam Email Classifier — sklearn Voting Ensemble (Gradio)

⚠️ DEPRECATED — This model has been merged into spam-xai-classifier-v2. v2 retrains the same RF + LR + SVM VotingClassifier ensemble on the full 99,999-sample corpus (vs. the smaller subset used here) and adds the LIME / SHAP / ELI5 XAI workflow. This repository is preserved for archival reference only — no further updates will be made. Please use v2 for new work.

ENGT 375 — Applied Machine Learning | Spring 2026 | ODU A voting ensemble classifier (Random Forest + Logistic Regression + SVM) for spam email detection, with LIME and SHAP explainability support.

Model Details

• Architecture: VotingClassifier (soft voting)
  • Random Forest
  • Logistic Regression
  • Calibrated LinearSVC
• Features: TF-IDF (text) + 24 hand-crafted metadata features
• Framework: scikit-learn
• Task: Binary classification (spam / ham)

Evaluation Results

Training Details

Dataset

• VoltageVagabond/spam-email-dataset
• Sources: Kaggle 190K spam/ham + GitHub email-dataset

Files

Usage

import joblib
from utils import preprocess_text, compute_metadata_features

model = joblib.load("voting_model.joblib")
tfidf = joblib.load("tfidf_vectorizer.joblib")
scaler = joblib.load("meta_scaler.joblib")
threshold = joblib.load("optimal_threshold.joblib")

email = "Congratulations! You've won a free iPhone!"
text_features = tfidf.transform([preprocess_text(email)])
meta_features = scaler.transform([compute_metadata_features(email)])
features = hstack([text_features, csr_matrix(meta_features)])

proba = model.predict_proba(features)[0][1]
label = "SPAM" if proba >= threshold else "HAM"

Interactive Demo

• Gradio Space

Intended Use

This model is an educational demonstration of sklearn ensemble methods with explainable AI (XAI), created as part of a university course project. It is suitable for:

• Learning how voting ensembles combine multiple classifiers
• Understanding TF-IDF text vectorization with metadata feature engineering
• Exploring LIME and SHAP explanations for model interpretability

It is not intended for production spam filtering.

Limitations

• Bag-of-words approach — TF-IDF cannot distinguish legitimate marketing from spam when vocabulary overlaps significantly
• Binary classification only (spam/ham) — no multi-class or severity ranking
• Trained on English emails only — not suitable for other languages
• Static vocabulary — cannot adapt to new spam patterns without retraining
• Threshold tuning is dataset-specific and may not generalize

Related Models

Citation

@misc{voltagevagabond2026spamgradio,
  title={Spam Email Classifier — sklearn Voting Ensemble (Gradio)},
  author={VoltageVagabond},
  year={2026},
  howpublished={\url{https://huggingface.co/VoltageVagabond/spam-classifier-gradio-model}},
  note={ENGT 375 — Applied Machine Learning, Old Dominion University, Spring 2026}
}