Skip to main content

Email classifier into 5 types using DistilBERT

Project description

📬 mail-bandit

mail-bandit is a plug-and-play email classifier that reads emails from an Amazon S3 bucket and classifies them into 5 categories using a fine-tuned DistilBERT model and XGBoost classifier.


🚀 Features

  • ✅ Classifies emails into:
    • Offer Rollout
    • Rejection
    • Additional Information Required
    • General Communication
    • Unrelated to Application
  • ✅ DistilBERT embeddings + XGBoost classifier
  • ✅ Reads email data from .jsonl stored in S3
  • ✅ Easy to train, deploy, and extend

🛠 Installation

pip install mail-bandit

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

mail_bandit-0.1.1.tar.gz (3.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

mail_bandit-0.1.1-py3-none-any.whl (4.4 kB view details)

Uploaded Python 3

File details

Details for the file mail_bandit-0.1.1.tar.gz.

File metadata

  • Download URL: mail_bandit-0.1.1.tar.gz
  • Upload date:
  • Size: 3.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.10

File hashes

Hashes for mail_bandit-0.1.1.tar.gz
Algorithm Hash digest
SHA256 b8e906a3002926f2eadc1149fdcfb4482c1b4bc93a657ca8dc8b020bea09ef32
MD5 aeea61a0c7abff55003df2c4a1264613
BLAKE2b-256 16583973e26f14b95b4ac365c455ece5ee89bdcbdfa293f02cb2e95efada0d07

See more details on using hashes here.

File details

Details for the file mail_bandit-0.1.1-py3-none-any.whl.

File metadata

  • Download URL: mail_bandit-0.1.1-py3-none-any.whl
  • Upload date:
  • Size: 4.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.10

File hashes

Hashes for mail_bandit-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 053788fcf7bef9a2f8cbdd999ab2664e0ae1cad9fc8d737da4c8245fe81da83c
MD5 7005ce584f5f950fff446e568d950538
BLAKE2b-256 04d38159358a21cf6bd7438dd061289006c366080d768cb951c34077d0168a2e

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page