Downloads · 30 days
30
25% of all-time downloads
polodealvarado/late_interaction
late_interaction is a zero-shot classification model from polodealvarado. Use it when you need labels you did not train the model on. It is set up for transformers. The card lists the license as mit.
ColBERT-style token-level MaxSim scoring for fine-grained alignment.
Downloads · 30 days
30
25% of all-time downloads
All-time downloads
118
Public
Parameters
110M
438 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors438 MB · 100%
From the Hugging Face model README
ColBERT-style token-level MaxSim scoring for fine-grained alignment.
This model encodes texts and candidate labels into a shared embedding space using BERT, enabling classification into arbitrary categories without retraining for new labels.
| Parameter | Value |
|---|---|
| Base model | bert-base-uncased |
| Model variant | late_interaction |
| Training steps | 1000 |
| Batch size | 2 |
| Learning rate | 2e-05 |
| Trainable params | 109,580,672 |
| Training time | 354.7s |
Trained on polodealvarado/zeroshot-classification.
| Metric | Score |
|---|---|
| Precision | 0.8546 |
| Recall | 0.9686 |
| F1 Score | 0.9081 |
from models.late_interaction import LateInteractionModel
model = LateInteractionModel.from_pretrained("polodealvarado/late_interaction")
predictions = model.predict(
texts=["The stock market crashed yesterday."],
labels=[["Finance", "Sports", "Biology", "Economy"]],
)
print(predictions)
# [{"text": "...", "scores": {"Finance": 0.98, "Economy": 0.85, ...}}]