setfit sentence-transformers text-classification

y3r/poc-mathmeow-classification

This is a SetFit model that can be used for text classification. The model has been trained using an efficient few-shot learning technique that involves:

  1. Fine-tuning a Sentence Transformer with contrastive learning.
  2. Training a classification head with features from the fine-tuned Sentence Transformer.

Usage

To use this model for inference, first install the SetFit library:

python -m pip install setfit

You can then run inference as follows:

from setfit import SetFitModel

# Download from Hub and run inference
model = SetFitModel.from_pretrained("y3r/poc-mathmeow-classification")
# Run inference
texts = [
    'คำนวณเรื่องดอกเบี้ยทบต้นคิดยังไง',
    'เรื่อง empty set'
    'complement มันคืออะไรเหรอ',
    'ใครก็ได้อธิบายเรื่องคอมพลีเมนต์ให้ที',
    'ตรรกศาสตร์คืออัลไร',
    'สอนเรื่อง intersec ให้หน่อย',
    'ที่ร้านมีที่จอดรถมั้ยคับ',
]
preds = model(texts)