Hugging Face's logo

yaniseuranova
/
setfit-rag-hybrid-search-query-router

Text Classification
setfit
Safetensors
sentence-transformers
xlm-roberta
generated_from_setfit_trainer
Eval Results
Model card Files Community
yaniseuranova commited on
Commit
a16f48d
•
1 Parent(s): 9067fbf

Add SetFit model

Browse files
Files changed (5) hide show
  1. README.md +34 -32
  2. config.json +1 -1
  3. config_setfit.json +2 -2
  4. model.safetensors +1 -1
  5. model_head.pkl +1 -1
README.md CHANGED
@@ -129,11 +129,11 @@ preds = model("How do societal norms influence the emergence of new business mod
129
  ### Training Set Metrics
130
  | Training set | Min | Median | Max |
131
  |:-------------|:----|:--------|:----|
132
- | Word count | 5 | 19.0729 | 56 |
133
 
134
  | Label | Training Sample Count |
135
  |:---------|:----------------------|
136
- | lexical | 44 |
137
  | semantic | 52 |
138
 
139
  ### Training Hyperparameters
@@ -154,36 +154,38 @@ preds = model("How do societal norms influence the emergence of new business mod
154
  - load_best_model_at_end: True
155
 
156
  ### Training Results
157
- | Epoch | Step | Training Loss | Validation Loss |
158
- |:-------:|:--------:|:-------------:|:---------------:|
159
- | 0.0034 | 1 | 0.2589 | - |
160
- | 0.1689 | 50 | 0.0421 | - |
161
- | 0.3378 | 100 | 0.0021 | - |
162
- | 0.5068 | 150 | 0.0006 | - |
163
- | 0.6757 | 200 | 0.0002 | - |
164
- | 0.8446 | 250 | 0.0002 | - |
165
- | 1.0 | 296 | - | 0.0001 |
166
- | 1.0135 | 300 | 0.0001 | - |
167
- | 1.1824 | 350 | 0.0001 | - |
168
- | 1.3514 | 400 | 0.0001 | - |
169
- | 1.5203 | 450 | 0.0001 | - |
170
- | 1.6892 | 500 | 0.0001 | - |
171
- | 1.8581 | 550 | 0.0001 | - |
172
- | 2.0 | 592 | - | 0.0 |
173
- | 2.0270 | 600 | 0.0 | - |
174
- | 2.1959 | 650 | 0.0001 | - |
175
- | 2.3649 | 700 | 0.0001 | - |
176
- | 2.5338 | 750 | 0.0001 | - |
177
- | 2.7027 | 800 | 0.0001 | - |
178
- | 2.8716 | 850 | 0.0001 | - |
179
- | 3.0 | 888 | - | 0.0 |
180
- | 3.0405 | 900 | 0.0 | - |
181
- | 3.2095 | 950 | 0.0001 | - |
182
- | 3.3784 | 1000 | 0.0001 | - |
183
- | 3.5473 | 1050 | 0.0001 | - |
184
- | 3.7162 | 1100 | 0.0001 | - |
185
- | 3.8851 | 1150 | 0.0001 | - |
186
- | **4.0** | **1184** | **-** | **0.0** |
 
 
187
 
188
  * The bold row denotes the saved checkpoint.
189
  ### Framework Versions
 
129
  ### Training Set Metrics
130
  | Training set | Min | Median | Max |
131
  |:-------------|:----|:--------|:----|
132
+ | Word count | 4 | 18.6566 | 56 |
133
 
134
  | Label | Training Sample Count |
135
  |:---------|:----------------------|
136
+ | lexical | 47 |
137
  | semantic | 52 |
138
 
139
  ### Training Hyperparameters
 
154
  - load_best_model_at_end: True
155
 
156
  ### Training Results
157
+ | Epoch | Step | Training Loss | Validation Loss |
158
+ |:-------:|:-------:|:-------------:|:---------------:|
159
+ | 0.0032 | 1 | 0.3177 | - |
160
+ | 0.1592 | 50 | 0.0905 | - |
161
+ | 0.3185 | 100 | 0.0013 | - |
162
+ | 0.4777 | 150 | 0.0011 | - |
163
+ | 0.6369 | 200 | 0.0002 | - |
164
+ | 0.7962 | 250 | 0.0003 | - |
165
+ | 0.9554 | 300 | 0.0001 | - |
166
+ | 1.0 | 314 | - | 0.0001 |
167
+ | 1.1146 | 350 | 0.0001 | - |
168
+ | 1.2739 | 400 | 0.0001 | - |
169
+ | 1.4331 | 450 | 0.0001 | - |
170
+ | 1.5924 | 500 | 0.0001 | - |
171
+ | 1.7516 | 550 | 0.0001 | - |
172
+ | 1.9108 | 600 | 0.0001 | - |
173
+ | 2.0 | 628 | - | 0.0 |
174
+ | 2.0701 | 650 | 0.0001 | - |
175
+ | 2.2293 | 700 | 0.0 | - |
176
+ | 2.3885 | 750 | 0.0001 | - |
177
+ | 2.5478 | 800 | 0.0 | - |
178
+ | 2.7070 | 850 | 0.0001 | - |
179
+ | 2.8662 | 900 | 0.0001 | - |
180
+ | **3.0** | **942** | **-** | **0.0** |
181
+ | 3.0255 | 950 | 0.0001 | - |
182
+ | 3.1847 | 1000 | 0.0 | - |
183
+ | 3.3439 | 1050 | 0.0001 | - |
184
+ | 3.5032 | 1100 | 0.0001 | - |
185
+ | 3.6624 | 1150 | 0.0 | - |
186
+ | 3.8217 | 1200 | 0.0001 | - |
187
+ | 3.9809 | 1250 | 0.0001 | - |
188
+ | 4.0 | 1256 | - | 0.0 |
189
 
190
  * The bold row denotes the saved checkpoint.
191
  ### Framework Versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "checkpoints/step_1184",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "checkpoints/step_942",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
config_setfit.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "normalize_embeddings": false,
3
  "labels": [
4
  "lexical",
5
  "semantic"
6
- ]
 
7
  }
 
1
  {
 
2
  "labels": [
3
  "lexical",
4
  "semantic"
5
+ ],
6
+ "normalize_embeddings": false
7
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:469d9abe15e2cc44f0345dc322550b58283c17df8066521df53bb703dd643b23
3
  size 437967672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbf6d3930a01c83d13c8e9917fadace89ffe82b20e0e8e1e79879f17c3f3ffea
3
  size 437967672
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:88fd4031a1711b1bb947166a88eef75fb197ace98bf0056e85781e9aa0e4ca96
3
  size 7039
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3eb7a8f7e177ed92e49ddc0fd6d56cff9748420121ded82c0713ee52aa1e2294
3
  size 7039