Downloads · 30 days
6
8% of all-time downloads
KingTechnician/osmosis-modernbert-joint
osmosis-modernbert-joint is a text classification model from KingTechnician. Use it when you need a label for a piece of text. It is set up for sentence-transformers. The card lists the license as mit.
Five-class response sufficiency classifier trained with SetFit-style contrastive learning on Yahoo Answers + Triage data.
Downloads · 30 days
6
8% of all-time downloads
All-time downloads
80
Public
Parameters
149M
596 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors596 MB · 99%
From the Hugging Face model README
Five-class response sufficiency classifier trained with SetFit-style contrastive learning on Yahoo Answers + Triage data.
| Evaluation | Accuracy | Macro F1 |
|---|---|---|
| Yahoo within-domain | 51.9% | 41.3% |
| Triage held-out | 94.7% | 94.7% |
| Class | F1 |
|---|---|
| ADDR_DIRECT | 93.0% |
| ADDR_PARTIAL | 94.2% |
| NOADDR_ON | 93.6% |
| NOADDR_TANGENTIAL | 92.5% |
| NOADDR_OFF | 100% |
from sentence_transformers import SentenceTransformer
import joblib
encoder = SentenceTransformer("KingTechnician/osmosis-modernbert-joint")
head = joblib.load("head.joblib") # download from this repo
text = "Objective:\nWhat causes rain?\nResponse: Rain forms when water vapor condenses."
embedding = encoder.encode([text])
prediction = head.predict(embedding)
labels = ["ADDR_DIRECT", "ADDR_PARTIAL", "NOADDR_ON", "NOADDR_TANGENTIAL", "NOADDR_OFF"]
print(labels[prediction[0]])