sauti-dispute-classifier

Это SetFit-модель для text-classification от автора sautiyadere на базе BAAI/bge-small-en-v1.5. Она относит сообщения водителей такси/райдхейлинга к одной из 4 категорий: fare_dispute, fraud_flag, passenger_safety и rating_or_cancellation. Размер — около 30 млн параметров, максимальная длина — 512 токенов.

📝 Ключев��е особенности

• 📝 SetFit: few-shot обучение через contrastive fine-tuning Sentence Transformer и LogisticRegression-голову.
• 🗣 4 класса для разбора конфликтных обращений водителей.
• ⌨ Базовая модель — BAAI/bge-small-en-v1.5, максимум 512 токенов.
• 👍 Обучение на 32 примерах: по 8 на каждый класс, с oversampling.
• 🤓 Запуск через библиотеку setfit в несколько строк.

📝 Пример

from setfit import SetFitModel

model = SetFitModel.from_pretrained('sautiyadere/sauti-dispute-classifier')
preds = model('deactivated low rating. started Saturday night. please help')
print(preds)

Подробнее

Валидационный loss — 0.1609 после 1 эпохи; обучение шло на 32 примерах, по 8 на каждый класс.
Голова — LogisticRegression, эмбеддинги — BAAI/bge-small-en-v1.5, максимум 512 токенов.
Бенчмарков и требований к VRAM в карточке нет; примеры в карточке — на английском и суахили.
Модель узкая и предназначена для классификации сообщений о спорах, а не для общей модерации.