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
4205681
•
1 Parent(s): 9eac52b

Add SetFit model

Browse files
Files changed (5) hide show
  1. README.md +20 -20
  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
@@ -130,8 +130,8 @@ preds = model("Comment rédiger un bon CV?")
130
 
131
  | Label | Training Sample Count |
132
  |:----------------------|:----------------------|
133
- | very_semantic_queries | 16 |
134
- | semantic_queries | 18 |
135
  | lexical_queries | 1 |
136
  | very_lexical | 15 |
137
 
@@ -155,24 +155,24 @@ preds = model("Comment rédiger un bon CV?")
155
  ### Training Results
156
  | Epoch | Step | Training Loss | Validation Loss |
157
  |:-------:|:-------:|:-------------:|:---------------:|
158
- | 0.0059 | 1 | 0.4006 | - |
159
- | 0.2941 | 50 | 0.1896 | - |
160
- | 0.5882 | 100 | 0.052 | - |
161
- | 0.8824 | 150 | 0.0042 | - |
162
- | 1.0 | 170 | - | 0.0023 |
163
- | 1.1765 | 200 | 0.0011 | - |
164
- | 1.4706 | 250 | 0.0006 | - |
165
- | 1.7647 | 300 | 0.0007 | - |
166
- | 2.0 | 340 | - | 0.0003 |
167
- | 2.0588 | 350 | 0.0004 | - |
168
- | 2.3529 | 400 | 0.0004 | - |
169
- | 2.6471 | 450 | 0.0004 | - |
170
- | 2.9412 | 500 | 0.0009 | - |
171
- | 3.0 | 510 | - | 0.0003 |
172
- | 3.2353 | 550 | 0.0003 | - |
173
- | 3.5294 | 600 | 0.0004 | - |
174
- | 3.8235 | 650 | 0.0003 | - |
175
- | **4.0** | **680** | **-** | **0.0002** |
176
 
177
  * The bold row denotes the saved checkpoint.
178
  ### Framework Versions
 
130
 
131
  | Label | Training Sample Count |
132
  |:----------------------|:----------------------|
133
+ | very_semantic_queries | 13 |
134
+ | semantic_queries | 21 |
135
  | lexical_queries | 1 |
136
  | very_lexical | 15 |
137
 
 
155
  ### Training Results
156
  | Epoch | Step | Training Loss | Validation Loss |
157
  |:-------:|:-------:|:-------------:|:---------------:|
158
+ | 0.0060 | 1 | 0.3581 | - |
159
+ | 0.2976 | 50 | 0.2142 | - |
160
+ | 0.5952 | 100 | 0.1448 | - |
161
+ | 0.8929 | 150 | 0.0094 | - |
162
+ | 1.0 | 168 | - | 0.0034 |
163
+ | 1.1905 | 200 | 0.0014 | - |
164
+ | 1.4881 | 250 | 0.0009 | - |
165
+ | 1.7857 | 300 | 0.0007 | - |
166
+ | 2.0 | 336 | - | 0.0003 |
167
+ | 2.0833 | 350 | 0.0012 | - |
168
+ | 2.3810 | 400 | 0.0006 | - |
169
+ | 2.6786 | 450 | 0.0004 | - |
170
+ | 2.9762 | 500 | 0.0005 | - |
171
+ | 3.0 | 504 | - | 0.0003 |
172
+ | 3.2738 | 550 | 0.0006 | - |
173
+ | 3.5714 | 600 | 0.0004 | - |
174
+ | 3.8690 | 650 | 0.0003 | - |
175
+ | **4.0** | **672** | **-** | **0.0002** |
176
 
177
  * The bold row denotes the saved checkpoint.
178
  ### Framework Versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "checkpoints/step_680",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "checkpoints/step_672",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
config_setfit.json CHANGED
@@ -1,9 +1,9 @@
1
  {
 
2
  "labels": [
3
  "very_semantic_queries",
4
  "semantic_queries",
5
  "lexical_queries",
6
  "very_lexical"
7
- ],
8
- "normalize_embeddings": false
9
  }
 
1
  {
2
+ "normalize_embeddings": false,
3
  "labels": [
4
  "very_semantic_queries",
5
  "semantic_queries",
6
  "lexical_queries",
7
  "very_lexical"
8
+ ]
 
9
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:90b2b4e9ab2e110c6b28701f1982fe75210d765ad667f050c5225fb0e218d56e
3
  size 437967672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7176f55cb144d7f23531b94ed150020322801f7e4507f17192447bbd8d7a497
3
  size 437967672
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1b648aa4a967e622bed4524cadcef504ec05d246c86c8783556500585c58c06
3
  size 38263
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70861c361e27e864a51a592de6f9d6069d09dd09a1605f3a871b8061625a56d1
3
  size 38263