Add SetFit model
Browse files- README.md +28 -15
- config_setfit.json +2 -2
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
@@ -74,7 +74,7 @@ model-index:
|
|
74 |
split: test
|
75 |
metrics:
|
76 |
- type: accuracy
|
77 |
-
value: 0.
|
78 |
name: Accuracy
|
79 |
---
|
80 |
|
@@ -116,7 +116,7 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
116 |
### Metrics
|
117 |
| Label | Accuracy |
|
118 |
|:--------|:---------|
|
119 |
-
| **all** | 0.
|
120 |
|
121 |
## Uses
|
122 |
|
@@ -180,7 +180,7 @@ Evaluation:")
|
|
180 |
|
181 |
### Training Hyperparameters
|
182 |
- batch_size: (16, 16)
|
183 |
-
- num_epochs: (
|
184 |
- max_steps: -1
|
185 |
- sampling_strategy: oversampling
|
186 |
- num_iterations: 20
|
@@ -201,19 +201,32 @@ Evaluation:")
|
|
201 |
| Epoch | Step | Training Loss | Validation Loss |
|
202 |
|:------:|:----:|:-------------:|:---------------:|
|
203 |
| 0.0015 | 1 | 0.2012 | - |
|
204 |
-
| 0.0746 | 50 | 0.
|
205 |
-
| 0.1493 | 100 | 0.
|
206 |
| 0.2239 | 150 | 0.2507 | - |
|
207 |
-
| 0.2985 | 200 | 0.
|
208 |
-
| 0.3731 | 250 | 0.
|
209 |
-
| 0.4478 | 300 | 0.
|
210 |
-
| 0.5224 | 350 | 0.
|
211 |
-
| 0.5970 | 400 | 0.
|
212 |
-
| 0.6716 | 450 | 0.
|
213 |
-
| 0.7463 | 500 | 0.
|
214 |
-
| 0.8209 | 550 | 0.
|
215 |
-
| 0.8955 | 600 | 0.
|
216 |
-
| 0.9701 | 650 | 0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
217 |
|
218 |
### Framework Versions
|
219 |
- Python: 3.10.14
|
|
|
74 |
split: test
|
75 |
metrics:
|
76 |
- type: accuracy
|
77 |
+
value: 0.676056338028169
|
78 |
name: Accuracy
|
79 |
---
|
80 |
|
|
|
116 |
### Metrics
|
117 |
| Label | Accuracy |
|
118 |
|:--------|:---------|
|
119 |
+
| **all** | 0.6761 |
|
120 |
|
121 |
## Uses
|
122 |
|
|
|
180 |
|
181 |
### Training Hyperparameters
|
182 |
- batch_size: (16, 16)
|
183 |
+
- num_epochs: (2, 2)
|
184 |
- max_steps: -1
|
185 |
- sampling_strategy: oversampling
|
186 |
- num_iterations: 20
|
|
|
201 |
| Epoch | Step | Training Loss | Validation Loss |
|
202 |
|:------:|:----:|:-------------:|:---------------:|
|
203 |
| 0.0015 | 1 | 0.2012 | - |
|
204 |
+
| 0.0746 | 50 | 0.2564 | - |
|
205 |
+
| 0.1493 | 100 | 0.2529 | - |
|
206 |
| 0.2239 | 150 | 0.2507 | - |
|
207 |
+
| 0.2985 | 200 | 0.2237 | - |
|
208 |
+
| 0.3731 | 250 | 0.1595 | - |
|
209 |
+
| 0.4478 | 300 | 0.0502 | - |
|
210 |
+
| 0.5224 | 350 | 0.0254 | - |
|
211 |
+
| 0.5970 | 400 | 0.0094 | - |
|
212 |
+
| 0.6716 | 450 | 0.0051 | - |
|
213 |
+
| 0.7463 | 500 | 0.0028 | - |
|
214 |
+
| 0.8209 | 550 | 0.003 | - |
|
215 |
+
| 0.8955 | 600 | 0.0022 | - |
|
216 |
+
| 0.9701 | 650 | 0.002 | - |
|
217 |
+
| 1.0448 | 700 | 0.0018 | - |
|
218 |
+
| 1.1194 | 750 | 0.0016 | - |
|
219 |
+
| 1.1940 | 800 | 0.0015 | - |
|
220 |
+
| 1.2687 | 850 | 0.0015 | - |
|
221 |
+
| 1.3433 | 900 | 0.0015 | - |
|
222 |
+
| 1.4179 | 950 | 0.0014 | - |
|
223 |
+
| 1.4925 | 1000 | 0.0014 | - |
|
224 |
+
| 1.5672 | 1050 | 0.0013 | - |
|
225 |
+
| 1.6418 | 1100 | 0.0013 | - |
|
226 |
+
| 1.7164 | 1150 | 0.0012 | - |
|
227 |
+
| 1.7910 | 1200 | 0.0012 | - |
|
228 |
+
| 1.8657 | 1250 | 0.0013 | - |
|
229 |
+
| 1.9403 | 1300 | 0.0012 | - |
|
230 |
|
231 |
### Framework Versions
|
232 |
- Python: 3.10.14
|
config_setfit.json
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
{
|
2 |
-
"
|
3 |
-
"
|
4 |
}
|
|
|
1 |
{
|
2 |
+
"normalize_embeddings": false,
|
3 |
+
"labels": null
|
4 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 437951328
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:142211a3b2913355b2acedf52733650ec37689b214590d3eb94b191b4a5d72b2
|
3 |
size 437951328
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 7007
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de66766034da388e376ba859ce844ed0fe0189eb827c1c91d9903b49dd6363c6
|
3 |
size 7007
|