McOwska's picture
Add trained model bundle
b786f3a verified

Action Requirement Classifier

This repository contains a lightweight text classifier for predicting whether a message is:

  • action_required
  • informational
  • optional_action

What’s inside

  • classifier.joblib: a scikit-learn classifier trained on SentenceTransformer embeddings
  • encoder/: the SentenceTransformer encoder saved via SentenceTransformer.save()
  • metadata.json: bundle metadata (model selection metric, encoder name, etc.)

Quickstart (Python)

import joblib
from sentence_transformers import SentenceTransformer

encoder = SentenceTransformer("./encoder")
clf = joblib.load("classifier.joblib")

texts = ["Please upload a clearer photo of your ID."]
emb = encoder.encode(texts)
pred = clf.predict(emb)
print(pred[0])

Notes

  • The classifier expects embeddings from the included encoder/ directory.
  • If you move files around, update paths accordingly.