ShiWarai commited on
Commit
8efb7bb
·
verified ·
1 Parent(s): 023dc4a

CI: модель обучена и загружена (21ce1c9)

Browse files
2_Dense/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ffbd357e603d1635ea0ffb287ac4d6b640e8410cef9392bbce696809229b2520
3
  size 9437272
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6aa91f3b5776c57517547ed18ead3a7c066dd39acfeca9702ba3d97a5abb2ed3
3
  size 9437272
3_Dense/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:144adf0dcbf3552f465aa5d893d42b0d81eac5c9e2d1dd8f4a8b0b354cbf88fb
3
  size 9437272
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cba00cf4accd6f28f14e1a1b9fe39f2285c35eff52db64112b72ecb51a8df18
3
  size 9437272
README.md CHANGED
@@ -5,11 +5,11 @@ tags:
5
  - text-classification
6
  - generated_from_setfit_trainer
7
  widget:
8
- - text: робот не любит бегать
9
- - text: надо равняться
10
- - text: панда ложись
11
- - text: ну остановись
12
- - text: беги бы
13
  metrics:
14
  - accuracy
15
  pipeline_tag: text-classification
@@ -28,7 +28,7 @@ model-index:
28
  split: test
29
  metrics:
30
  - type: accuracy
31
- value: 0.890282131661442
32
  name: Accuracy
33
  ---
34
 
@@ -48,7 +48,7 @@ The model has been trained using an efficient few-shot learning technique that i
48
  - **Sentence Transformer body:** [google/embeddinggemma-300M](https://huggingface.co/google/embeddinggemma-300M)
49
  - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
50
  - **Maximum Sequence Length:** 2048 tokens
51
- - **Number of Classes:** 13 classes
52
  <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
53
  <!-- - **Language:** Unknown -->
54
  <!-- - **License:** Unknown -->
@@ -60,28 +60,18 @@ The model has been trained using an efficient few-shot learning technique that i
60
  - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
61
 
62
  ### Model Labels
63
- | Label | Examples |
64
- |:-------------------|:------------------------------------------------------------------------------------------------------------------------------|
65
- | unknown | <ul><li>'чей робот'</li><li>'опять лежать'</li><li>'смотри панда'</li></ul> |
66
- | stand_at_attention | <ul><li>'пора выравняться'</li><li>'не хочешь равняться'</li><li>'выравнялся бы'</li></ul> |
67
- | dismiss | <ul><li>'давай поднимись'</li><li>'пора подняться'</li><li>'эй встань'</li></ul> |
68
- | silence | <ul><li>'перестать говорить'</li><li>'перестаньте болтать'</li><li>'замолкать'</li></ul> |
69
- | rotate | <ul><li>'переворачиваться'</li><li>'кувыркнись'</li><li>'ты кувыркайся'</li></ul> |
70
- | give_paw | <ul><li>'ну лапу дай'</li><li>'хотела бы чтобы панда дала лапу'</li><li>'панда давай лапу'</li></ul> |
71
- | stop_running | <ul><li>'а ну остановись'</li><li>'смирно'</li><li>'попроси робота остановиться'</li></ul> |
72
- | reconnect_joystick | <ul><li>'не хочешь подключить джойстик'</li><li>'подключись к джойстику сейчас'</li><li>'можно подключить джойстик'</li></ul> |
73
- | bind | <ul><li>'привязывать робота'</li><li>'панда привяжи панду'</li><li>'привяжите робота'</li></ul> |
74
- | unbind | <ul><li>'панда отвяжи панду'</li><li>'отвяжешь панду'</li><li>'отвяжи панду панда'</li></ul> |
75
- | lie_down | <ul><li>'быстро ложись'</li><li>'упасть'</li><li>'полежи'</li></ul> |
76
- | run | <ul><li>'надо бежать'</li><li>'побеги'</li><li>'хотела бы чтобы панда бежала'</li></ul> |
77
- | help | <ul><li>'надо помочь'</li><li>'помог бы'</li><li>'команды'</li></ul> |
78
 
79
  ## Evaluation
80
 
81
  ### Metrics
82
  | Label | Accuracy |
83
  |:--------|:---------|
84
- | **all** | 0.8903 |
85
 
86
  ## Uses
87
 
@@ -99,9 +89,9 @@ Then you can load this model and run inference.
99
  from setfit import SetFitModel
100
 
101
  # Download from the 🤗 Hub
102
- model = SetFitModel.from_pretrained("tmp4koqu5h5/panda_commands")
103
  # Run inference
104
- preds = model("беги бы")
105
  ```
106
 
107
  <!--
@@ -133,26 +123,16 @@ preds = model("беги бы")
133
  ### Training Set Metrics
134
  | Training set | Min | Median | Max |
135
  |:-------------|:----|:-------|:----|
136
- | Word count | 1 | 2.3697 | 7 |
137
-
138
- | Label | Training Sample Count |
139
- |:-------------------|:----------------------|
140
- | bind | 44 |
141
- | dismiss | 128 |
142
- | give_paw | 83 |
143
- | help | 18 |
144
- | lie_down | 88 |
145
- | reconnect_joystick | 108 |
146
- | rotate | 109 |
147
- | run | 85 |
148
- | silence | 22 |
149
- | stand_at_attention | 70 |
150
- | stop_running | 108 |
151
- | unbind | 30 |
152
- | unknown | 381 |
153
 
154
  ### Training Hyperparameters
155
- - batch_size: (128, 128)
156
  - num_epochs: (1, 1)
157
  - max_steps: -1
158
  - sampling_strategy: oversampling
@@ -173,19 +153,12 @@ preds = model("беги бы")
173
  ### Training Results
174
  | Epoch | Step | Training Loss | Validation Loss |
175
  |:------:|:----:|:-------------:|:---------------:|
176
- | 0.0025 | 1 | 0.2328 | - |
177
- | 0.1253 | 50 | 0.0955 | - |
178
- | 0.2506 | 100 | 0.0228 | - |
179
- | 0.3759 | 150 | 0.0098 | - |
180
- | 0.5013 | 200 | 0.0044 | - |
181
- | 0.6266 | 250 | 0.0031 | - |
182
- | 0.7519 | 300 | 0.0031 | - |
183
- | 0.8772 | 350 | 0.0022 | - |
184
 
185
  ### Framework Versions
186
  - Python: 3.11.14
187
  - SetFit: 1.1.3
188
- - Sentence Transformers: 5.2.1
189
  - Transformers: 4.57.6
190
  - PyTorch: 2.9.1+cu128
191
  - Datasets: 4.5.0
 
5
  - text-classification
6
  - generated_from_setfit_trainer
7
  widget:
8
+ - text: прекрати
9
+ - text: ну
10
+ - text: приляг
11
+ - text: не понял
12
+ - text: отдохни
13
  metrics:
14
  - accuracy
15
  pipeline_tag: text-classification
 
28
  split: test
29
  metrics:
30
  - type: accuracy
31
+ value: 0.6666666666666666
32
  name: Accuracy
33
  ---
34
 
 
48
  - **Sentence Transformer body:** [google/embeddinggemma-300M](https://huggingface.co/google/embeddinggemma-300M)
49
  - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
50
  - **Maximum Sequence Length:** 2048 tokens
51
+ - **Number of Classes:** 3 classes
52
  <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
53
  <!-- - **Language:** Unknown -->
54
  <!-- - **License:** Unknown -->
 
60
  - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
61
 
62
  ### Model Labels
63
+ | Label | Examples |
64
+ |:---------|:----------------------------------------------------------------|
65
+ | dismiss | <ul><li>'хватит'</li><li>'стоп'</li><li>'остановись'</li></ul> |
66
+ | unknown | <ul><li>'не знаю'</li><li>'э'</li><li>'зачем'</li></ul> |
67
+ | lie_down | <ul><li>'ложись спать'</li><li>'ляг'</li><li>'лягись'</li></ul> |
 
 
 
 
 
 
 
 
 
 
68
 
69
  ## Evaluation
70
 
71
  ### Metrics
72
  | Label | Accuracy |
73
  |:--------|:---------|
74
+ | **all** | 0.6667 |
75
 
76
  ## Uses
77
 
 
89
  from setfit import SetFitModel
90
 
91
  # Download from the 🤗 Hub
92
+ model = SetFitModel.from_pretrained("tmpyksswlz9/panda_commands")
93
  # Run inference
94
+ preds = model("ну")
95
  ```
96
 
97
  <!--
 
123
  ### Training Set Metrics
124
  | Training set | Min | Median | Max |
125
  |:-------------|:----|:-------|:----|
126
+ | Word count | 1 | 1.1667 | 2 |
127
+
128
+ | Label | Training Sample Count |
129
+ |:---------|:----------------------|
130
+ | dismiss | 8 |
131
+ | lie_down | 8 |
132
+ | unknown | 8 |
 
 
 
 
 
 
 
 
 
 
133
 
134
  ### Training Hyperparameters
135
+ - batch_size: (32, 32)
136
  - num_epochs: (1, 1)
137
  - max_steps: -1
138
  - sampling_strategy: oversampling
 
153
  ### Training Results
154
  | Epoch | Step | Training Loss | Validation Loss |
155
  |:------:|:----:|:-------------:|:---------------:|
156
+ | 0.0333 | 1 | 0.23 | - |
 
 
 
 
 
 
 
157
 
158
  ### Framework Versions
159
  - Python: 3.11.14
160
  - SetFit: 1.1.3
161
+ - Sentence Transformers: 5.2.2
162
  - Transformers: 4.57.6
163
  - PyTorch: 2.9.1+cu128
164
  - Datasets: 4.5.0
config_sentence_transformers.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "model_type": "SentenceTransformer",
3
  "__version__": {
4
- "sentence_transformers": "5.2.1",
5
  "transformers": "4.57.6",
6
  "pytorch": "2.9.1+cu128"
7
  },
 
1
  {
2
  "model_type": "SentenceTransformer",
3
  "__version__": {
4
+ "sentence_transformers": "5.2.2",
5
  "transformers": "4.57.6",
6
  "pytorch": "2.9.1+cu128"
7
  },
config_setfit.json CHANGED
@@ -1,17 +1,7 @@
1
  {
2
  "labels": [
3
- "bind",
4
  "dismiss",
5
- "give_paw",
6
- "help",
7
  "lie_down",
8
- "reconnect_joystick",
9
- "rotate",
10
- "run",
11
- "silence",
12
- "stand_at_attention",
13
- "stop_running",
14
- "unbind",
15
  "unknown"
16
  ],
17
  "normalize_embeddings": false
 
1
  {
2
  "labels": [
 
3
  "dismiss",
 
 
4
  "lie_down",
 
 
 
 
 
 
 
5
  "unknown"
6
  ],
7
  "normalize_embeddings": false
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a48b877b051dfd0d0f3db30d4902e9f7b400ad1f4baac77157f8af3d385b41d
3
  size 1211486072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:983d1c1e1805463b5b3549f2e6b9a588f54e9c25ad976ce5818b6a4e76f97c22
3
  size 1211486072
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4c7c402de572d9f0275e22adf98c74ad367409a904d425b229ea42b7d01888d0
3
- size 81743
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2468fd91f013b49270fb040c0e16dd32b592eb65fc19b56ae39c2f36b8330ca4
3
+ size 19375