CoolFace
Modelpublic

imnim/multi-label-email-classifier

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes5downloads
Model Card

Model Card for Model ID

<!-- Provide a quick summary of what the model is/does. --> Model is finetuned for the task of email labelling. It labels the given email into one or more than one categories based on email subject and email body.

Model Details

Model Description

<!-- Provide a longer summary of what this model is. --> The model classifies emails into the following 10 categories: "Business", "Personal", "Promotions", "Customer Support", "Job Application", "Finance & Bills", "Events & Invitations", "Travel & Bookings", "Reminders", "Newsletters"

I have prepared a synthetic but realistic dataset of 2,105 labeled emails. Each email includes a subject, body, and one or more categories.

  • —Developed by: imnim
  • —Model type: text-to-text
  • —Language(s) (NLP): English
  • —Finetuned from model: Llama-3.1-8B-Instruct

Model Sources

<!-- Provide the basic links for the model. -->

  • —Repository: https://github.com/contributerMe/multi-label-email-classifier
  • —Demo: https://huggingface.co/spaces/imnim/Multi-labelEmailClassifier

Technical Specifications

Model Architecture and Objective

Auto-regressive language model that uses an optimized transformer architecture.

Compute Infrastructure

Kaggle Notebook

Hardware

Trained on Kaggle's P100 GPU

Framework versions

  • —PEFT 0.15.2