Add SetFit model

README.md

@@ -5,11 +5,11 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: Кто претендует на предоставление общежития
-- text: На сколько специальностей можно подать документы?
-- text: Направления подготовки в вуцНа каком курсе начинается Военная кафедра
-- text: Приемная кампания
-- text: Я после колледжа. Как мне поступить?
+- text: Военная кафедра
+- text: Какие льготы есть у выпускников колледжа?
+- text: Какие этапы включает приемная кампания?
+- text: Сколько продлится приемная кампания
+- text: Какие требования для поступления в ВУЦ?
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -28,7 +28,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.8
+      value: 1.0
       name: Accuracy
 ---
 
@@ -48,7 +48,7 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Sentence Transformer body:** [cointegrated/rubert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2)
 - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
 - **Maximum Sequence Length:** 2048 tokens
-- **Number of Classes:** 8 classes
+- **Number of Classes:** 7 classes
 <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
 <!-- - **Language:** Unknown -->
 <!-- - **License:** Unknown -->
@@ -60,23 +60,22 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
 ### Model Labels
-| Label | Examples                                                                                                                                                                                            |
-|:------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| 2     | <ul><li>'Как получить дополнительные баллы для поступленияКак получить дополнительные баллы'</li><li>'Как получить доп баллы'</li><li>'Расскажи о допполнительных баллах при поступлении'</li></ul> |
-| 6     | <ul><li>'Какие есть вступительные испытания?'</li><li>'Внутренние экзамены для поступления'</li><li>'Экзамены для поступления'</li></ul>                                                            |
-| 1     | <ul><li>'Какие документы нужны для поступления'</li><li>'Необходимые документы для поступления'</li><li>'Документы на поступление каким способом подать можно?'</li></ul>                           |
-| 0     | <ul><li>'Приемная кампания в вузе'</li><li>'Как долго длится приемная кампания'</li><li>'Длительность приемной кампании'</li></ul>                                                                  |
-| 4     | <ul><li>'Военный учебный центр'</li><li>'Программы подготовки в ВУЦ'</li><li>'Как поступить на военную кафедру'</li></ul>                                                                           |
-| 7     | <ul><li>'Кто может жить в общежитии?'</li><li>'Как попасть в общагу?Как попасть в общежитии?'</li><li>'Кто может жить в общаге?'</li></ul>                                                          |
-| 5     | <ul><li>'Как поступить в вуз после колледжа?'</li><li>'Как поступить в Вуз после колледжа'</li><li>''</li></ul>                                                                                     |
-| 3     | <ul><li>'На сколько направлений подготовки можно подать документы?'</li><li>'На сколько специальностей можно подать документы?'</li></ul>                                                           |
+| Label | Examples                                                                                                                                                                          |
+|:------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| 3     | <ul><li>'Есть ли военная кафедра'</li><li>'Какие документы нужны для поступления в ВУЦ?'</li><li>'Как проходит отбор на военную кафедру'</li></ul>                                |
+| 2     | <ul><li>'Как подтвердить свои достижения?'</li><li>'Расскажи о дополнительных баллах при поступлении'</li><li>'Можно ли получить баллы за волонтерскую деятельность?'</li></ul>   |
+| 1     | <ul><li>'Нужно ли заверять копии документов?'</li><li>'Как узнать, что мои документы приняты?'</li><li>'Можно ли подать документы по почте?'</li></ul>                            |
+| 0     | <ul><li>'Как долго длится приемная кампания'</li><li>'Продолжительность приемной кампании'</li><li>'Когда заканчивается приемная кампания?'</li></ul>                             |
+| 4     | <ul><li>'Какие документы нужны для поступления после колледжа?'</li><li>'Как поступить в Вуз после колледжа'</li><li>'Какие сроки подачи документов после колледжа?'</li></ul>    |
+| 6     | <ul><li>'Как подать заявление на общежитие?'</li><li>'Кто претендует на предоставление общежития'</li><li>'Как оплачивать проживание в общежитии?'</li></ul>                      |
+| 5     | <ul><li>'Какие ошибки чаще всего допускают на экзаменах?'</li><li>'Какие вступительные испытания нужно сдавать?'</li><li>'Как подготовиться к вступительным экзаменам?'</li></ul> |
 
 ## Evaluation
 
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.8      |
+| **all** | 1.0      |
 
 ## Uses
 
@@ -96,7 +95,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("Maxim01/Intent_Classification_Test")
 # Run inference
-preds = model("Приемная кампания")
+preds = model("Военная кафедра")
 ```
 
 <!--
@@ -128,18 +127,17 @@ preds = model("Приемная кампания")
 ### Training Set Metrics
 | Training set | Min | Median | Max |
 |:-------------|:----|:-------|:----|
-| Word count   | 1   | 4.4103 | 9   |
+| Word count   | 2   | 5.3333 | 8   |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 6                     |
-| 1     | 8                     |
-| 2     | 5                     |
-| 3     | 2                     |
-| 4     | 5                     |
-| 5     | 4                     |
-| 6     | 4                     |
-| 7     | 5                     |
+| 0     | 9                     |
+| 1     | 11                    |
+| 2     | 6                     |
+| 3     | 9                     |
+| 4     | 10                    |
+| 5     | 11                    |
+| 6     | 10                    |
 
 ### Training Hyperparameters
 - batch_size: (8, 8)
@@ -163,10 +161,13 @@ preds = model("Приемная кампания")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0051 | 1    | 0.1228        | -               |
-| 0.2564 | 50   | 0.1409        | -               |
-| 0.5128 | 100  | 0.0783        | -               |
-| 0.7692 | 150  | 0.0682        | -               |
+| 0.0030 | 1    | 0.1199        | -               |
+| 0.1515 | 50   | 0.1727        | -               |
+| 0.3030 | 100  | 0.0936        | -               |
+| 0.4545 | 150  | 0.0599        | -               |
+| 0.6061 | 200  | 0.0529        | -               |
+| 0.7576 | 250  | 0.0436        | -               |
+| 0.9091 | 300  | 0.0359        | -               |
 
 ### Framework Versions
 - Python: 3.11.12

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:78aa50db3e67e2f1daf54fa23af99a5a19ca75e60a5c93ec302490990b21a8a9
+oid sha256:d271a20b8e11544b7421a9d88d8fec1b708e883606980c872708a9f418d5bd18
 size 116781184

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f4ab384aadb3f04c0bf18b03f949518d561c65e775b100ba9eb160ebff529a9d
-size 20935
+oid sha256:7b75e9934daa33dc331b30ae219a0a29096ddb1861635a87caa4e0092c9be5ae
+size 18431