Add SetFit model

README.md

@@ -5,11 +5,12 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: Сколько продлится приемная кампания?
-- text: Назови вступительные испытания
-- text: Как можно подать документы?
-- text: Какие документы нужны для поступления?
-- text: Как долго длится приемная кампания
+- text: Экзамены для поступления
+- text: Какие документы нужны для поступления
+- text: Продолжительность приемной кампании
+- text: Общежитие
+- text: Как получить дополнительные баллы для поступленияКак получить дополнительные
+    баллы
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -48,7 +49,7 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Sentence Transformer body:** [sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2](https://huggingface.co/sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2)
 - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
 - **Maximum Sequence Length:** 128 tokens
-- **Number of Classes:** 9 classes
+- **Number of Classes:** 8 classes
 <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
 <!-- - **Language:** Unknown -->
 <!-- - **License:** Unknown -->
@@ -60,17 +61,16 @@ 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>                           |
-| 4     | <ul><li>'На сколько специальностей можно подать документы?'</li><li>'На сколько направлений подготовки можно подать документы?'</li><li>'На сколько специальностей можно подать документы?'</li></ul> |
-| 7     | <ul><li>'Внутренние экзамены для поступления'</li><li>'Назови вступительные испытания'</li><li>'Какие есть вступительные испытания?'</li></ul>                                                        |
-| 5     | <ul><li>'Если я подам раньше оригиналы, то меня точно зачислят?'</li><li>'Имеют ли больше шансов те абитуриенты, кто подал документы и их оригиналы в первую очередь?'</li></ul>                      |
-| 1     | <ul><li>'Сколько бюджетных мест доступно в ОмГТУ?'</li><li>'Количество бюджетных мест в ОмГТУ?'</li></ul>                                                                                             |
-| 8     | <ul><li>'Как попасть в общагу?'</li><li>'Кто имеет право на получение общежития?'</li><li>'Кто может жить в общежитии?'</li></ul>                                                                     |
-| 6     | <ul><li>'Как поступить в вуз после колледжа?'</li><li>'Я после колледжа. Как мне поступить?'</li><li>'Послк колледжа'</li></ul>                                                                       |
-| 0     | <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>       |
+| 5     | <ul><li>'Я после колледжа. Как мне поступить?'</li><li>'Вуз после колледжа'</li><li>'Как поступить в вуз после колледжа?'</li></ul>                                                                 |
+| 7     | <ul><li>'Кто претендует на предоставление общежития'</li><li>'Общага'</li><li>'Кто может жить в общежитии?'</li></ul>                                                                               |
+| 6     | <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>                                         |
+| 2     | <ul><li>'Как получить дополнительные баллы для поступленияКак получить дополнительные баллы'</li><li>'Как получить доп баллы'</li><li>'Расскажи о допполнительных баллах при поступлении'</li></ul> |
+| 1     | <ul><li>'Какие документы нужны для поступления'</li><li>'Необходимые документы для поступления'</li><li>'Документы на поступление каким способом подать можно?'</li></ul>                           |
 
 ## Evaluation
 
@@ -97,7 +97,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("Maxim01/Intent_Classification_Test")
 # Run inference
-preds = model("Как можно подать документы?")
+preds = model("Общежитие")
 ```
 
 <!--
@@ -129,22 +129,21 @@ preds = model("Как можно подать документы?")
 ### Training Set Metrics
 | Training set | Min | Median | Max |
 |:-------------|:----|:-------|:----|
-| Word count   | 2   | 5.3103 | 15  |
+| Word count   | 1   | 4.3878 | 9   |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 5                     |
-| 1     | 2                     |
+| 0     | 7                     |
+| 1     | 8                     |
 | 2     | 5                     |
-| 3     | 2                     |
-| 4     | 3                     |
-| 5     | 2                     |
-| 6     | 3                     |
-| 7     | 3                     |
-| 8     | 4                     |
+| 3     | 4                     |
+| 4     | 7                     |
+| 5     | 6                     |
+| 6     | 5                     |
+| 7     | 7                     |
 
 ### Training Hyperparameters
-- batch_size: (16, 16)
+- batch_size: (8, 8)
 - num_epochs: (1, 1)
 - max_steps: -1
 - sampling_strategy: oversampling
@@ -165,8 +164,11 @@ preds = model("Как можно подать документы?")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0137 | 1    | 0.1213        | -               |
-| 0.6849 | 50   | 0.0317        | -               |
+| 0.0041 | 1    | 0.0986        | -               |
+| 0.2041 | 50   | 0.0729        | -               |
+| 0.4082 | 100  | 0.016         | -               |
+| 0.6122 | 150  | 0.0063        | -               |
+| 0.8163 | 200  | 0.0031        | -               |
 
 ### Framework Versions
 - Python: 3.11.11

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:21daefde02963c0796f5a984993b5b8ce8fc918479335f4d8f8921a294a7a5ca
+oid sha256:5907423a9906f5255b78828184b6eafc33d4b5a003620ec1aa2b6938f8321939
 size 470637416

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:267d4ed4c6a8b407bec08eb769cd81e29962f0a2e1b33ec377a12a1214b4a23e
-size 28639
+oid sha256:d655cb90ee7e60bbddf7e606591bed67a2b4d58dc5a4cd9ab5a25af033709ba7
+size 25543