Modelpublic

connect211/category-of-needs-classification-minilm-v3

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face

0likes25downloads

Model Card

category-of-needs-classification-minilm-v3

SetFit multi-label classifier on paraphrase-multilingual-MiniLM-L12-v2 for Connect211 search queries.

Taxonomy: UWWW Level-1 (19 categories). Replaces the Gravity L2 (79-label) minilm-v2 model.

Labels

Addictions
Arts, Culture and Recreation
Clothing/Personal/Household Needs
Disaster Services
Domestic Violence / Personal Safety
Education
Employment
Food/Meals
Health Care
Housing
Income Support/Assistance
Individual, Family and Community Support
Information Services
Legal, Consumer and Public Safety
Mental Health
Other Governmental/Economic Services
Transportation
Utilities
Volunteer & Donation

Usage

python

from setfit import SetFitModel

model = SetFitModel.from_pretrained("connect211/category-of-needs-classification-minilm-v3")
preds = model.predict(["I need help paying my rent", "food bank near me"])
print(preds)