PyXA script to classify email subject lines as junk or other using Latent Semantic Mapping

junk_email_classifier.py

import PyXA # Version 0.2.0
app = PyXA.Application("Mail")

dataset = {
    "junk": app.accounts()[0].mailboxes().by_name("Junk").messages().subject(),
    "other": app.accounts()[0].mailboxes().by_name("INBOX").messages().subject()
}

queries = [
    "Amazon Web Services Billing Statement Available",
    "Complete my registration form asap receive your rewards"
]

lsm = PyXA.XALSM(dataset)
for query in queries:
    category = list(dataset.keys())[lsm.categorize_query(query)[0][0] - 1]
    print(query, "- category:", category)
# Amazon Web Services Billing Statement Available - category: other
# Complete my registration form asap receive your rewards - category: junk