Add SetFit model
Browse files- README.md +25 -13
- config_setfit.json +2 -2
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
@@ -180,7 +180,7 @@ Final 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,18 +201,30 @@ Final evaluation:")
|
|
201 |
| Epoch | Step | Training Loss | Validation Loss |
|
202 |
|:------:|:----:|:-------------:|:---------------:|
|
203 |
| 0.0016 | 1 | 0.2275 | - |
|
204 |
-
| 0.0820 | 50 | 0.
|
205 |
-
| 0.1639 | 100 | 0.
|
206 |
-
| 0.2459 | 150 | 0.
|
207 |
-
| 0.3279 | 200 | 0.
|
208 |
-
| 0.4098 | 250 | 0.
|
209 |
-
| 0.4918 | 300 | 0.
|
210 |
-
| 0.5738 | 350 | 0.
|
211 |
-
| 0.6557 | 400 | 0.
|
212 |
-
| 0.7377 | 450 | 0.
|
213 |
-
| 0.8197 | 500 | 0.
|
214 |
-
| 0.9016 | 550 | 0.
|
215 |
-
| 0.9836 | 600 | 0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
216 |
|
217 |
### Framework Versions
|
218 |
- Python: 3.10.14
|
|
|
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.0016 | 1 | 0.2275 | - |
|
204 |
+
| 0.0820 | 50 | 0.2565 | - |
|
205 |
+
| 0.1639 | 100 | 0.2275 | - |
|
206 |
+
| 0.2459 | 150 | 0.1873 | - |
|
207 |
+
| 0.3279 | 200 | 0.1281 | - |
|
208 |
+
| 0.4098 | 250 | 0.0495 | - |
|
209 |
+
| 0.4918 | 300 | 0.0251 | - |
|
210 |
+
| 0.5738 | 350 | 0.0142 | - |
|
211 |
+
| 0.6557 | 400 | 0.0181 | - |
|
212 |
+
| 0.7377 | 450 | 0.0188 | - |
|
213 |
+
| 0.8197 | 500 | 0.0111 | - |
|
214 |
+
| 0.9016 | 550 | 0.0098 | - |
|
215 |
+
| 0.9836 | 600 | 0.0111 | - |
|
216 |
+
| 1.0656 | 650 | 0.0108 | - |
|
217 |
+
| 1.1475 | 700 | 0.0135 | - |
|
218 |
+
| 1.2295 | 750 | 0.0102 | - |
|
219 |
+
| 1.3115 | 800 | 0.0119 | - |
|
220 |
+
| 1.3934 | 850 | 0.0086 | - |
|
221 |
+
| 1.4754 | 900 | 0.0085 | - |
|
222 |
+
| 1.5574 | 950 | 0.0089 | - |
|
223 |
+
| 1.6393 | 1000 | 0.0101 | - |
|
224 |
+
| 1.7213 | 1050 | 0.0121 | - |
|
225 |
+
| 1.8033 | 1100 | 0.0112 | - |
|
226 |
+
| 1.8852 | 1150 | 0.0122 | - |
|
227 |
+
| 1.9672 | 1200 | 0.0099 | - |
|
228 |
|
229 |
### Framework Versions
|
230 |
- 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:4cdc1f7dc69d2bff56bffe2c0c077f71e2fa2d1fc56aa5c1a6acd8ccce9c412b
|
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:f2fe3f87f00406c241edf8bee4e92c3fc324e086ae51b41967ff729cdbf59009
|
3 |
size 7007
|