Add SetFit model
Browse files- 1_Pooling/config.json +1 -1
- README.md +42 -42
- config_setfit.json +2 -2
- model.safetensors +1 -1
- model_head.pkl +1 -1
1_Pooling/config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"word_embedding_dimension":
|
3 |
"pooling_mode_cls_token": false,
|
4 |
"pooling_mode_mean_tokens": true,
|
5 |
"pooling_mode_max_tokens": false,
|
|
|
1 |
{
|
2 |
+
"word_embedding_dimension": 768,
|
3 |
"pooling_mode_cls_token": false,
|
4 |
"pooling_mode_mean_tokens": true,
|
5 |
"pooling_mode_max_tokens": false,
|
README.md
CHANGED
@@ -309,47 +309,47 @@ preds = model("It is apparent that the predicted pressure trace for the advanced
|
|
309 |
### Training Results
|
310 |
| Epoch | Step | Training Loss | Validation Loss |
|
311 |
|:------:|:----:|:-------------:|:---------------:|
|
312 |
-
| 0.0005 | 1 | 0.
|
313 |
-
| 0.0241 | 50 | 0.
|
314 |
-
| 0.0482 | 100 | 0.
|
315 |
-
| 0.0723 | 150 | 0.
|
316 |
-
| 0.0964 | 200 | 0.
|
317 |
-
| 0.1205 | 250 | 0.
|
318 |
-
| 0.1446 | 300 | 0.
|
319 |
-
| 0.1688 | 350 | 0.
|
320 |
-
| 0.1929 | 400 | 0.
|
321 |
-
| 0.2170 | 450 | 0.
|
322 |
-
| 0.2411 | 500 | 0.
|
323 |
-
| 0.2652 | 550 | 0.
|
324 |
-
| 0.2893 | 600 | 0.
|
325 |
-
| 0.3134 | 650 | 0.
|
326 |
-
| 0.3375 | 700 | 0.
|
327 |
-
| 0.3616 | 750 | 0.
|
328 |
-
| 0.3857 | 800 | 0.
|
329 |
-
| 0.4098 | 850 | 0.
|
330 |
-
| 0.4339 | 900 | 0.
|
331 |
-
| 0.4581 | 950 | 0.
|
332 |
-
| 0.4822 | 1000 | 0.
|
333 |
-
| 0.5063 | 1050 | 0.
|
334 |
-
| 0.5304 | 1100 | 0.
|
335 |
-
| 0.5545 | 1150 | 0.
|
336 |
-
| 0.5786 | 1200 | 0.
|
337 |
-
| 0.6027 | 1250 | 0.
|
338 |
-
| 0.6268 | 1300 | 0.
|
339 |
-
| 0.6509 | 1350 | 0.
|
340 |
-
| 0.6750 | 1400 | 0.
|
341 |
-
| 0.6991 | 1450 | 0.
|
342 |
-
| 0.7232 | 1500 | 0.
|
343 |
-
| 0.7473 | 1550 | 0.
|
344 |
-
| 0.7715 | 1600 | 0.
|
345 |
-
| 0.7956 | 1650 | 0.
|
346 |
-
| 0.8197 | 1700 | 0.
|
347 |
-
| 0.8438 | 1750 | 0.
|
348 |
-
| 0.8679 | 1800 | 0.
|
349 |
-
| 0.8920 | 1850 | 0.
|
350 |
-
| 0.9161 | 1900 | 0.
|
351 |
-
| 0.9402 | 1950 | 0.
|
352 |
-
| 0.9643 | 2000 | 0.
|
353 |
| 0.9884 | 2050 | 0.0036 | - |
|
354 |
|
355 |
### Framework Versions
|
@@ -358,7 +358,7 @@ preds = model("It is apparent that the predicted pressure trace for the advanced
|
|
358 |
- Sentence Transformers: 2.2.2
|
359 |
- Transformers: 4.35.2
|
360 |
- PyTorch: 2.1.0+cu121
|
361 |
-
- Datasets: 2.
|
362 |
- Tokenizers: 0.15.0
|
363 |
|
364 |
## Citation
|
|
|
309 |
### Training Results
|
310 |
| Epoch | Step | Training Loss | Validation Loss |
|
311 |
|:------:|:----:|:-------------:|:---------------:|
|
312 |
+
| 0.0005 | 1 | 0.2624 | - |
|
313 |
+
| 0.0241 | 50 | 0.2244 | - |
|
314 |
+
| 0.0482 | 100 | 0.199 | - |
|
315 |
+
| 0.0723 | 150 | 0.2198 | - |
|
316 |
+
| 0.0964 | 200 | 0.1421 | - |
|
317 |
+
| 0.1205 | 250 | 0.0992 | - |
|
318 |
+
| 0.1446 | 300 | 0.1346 | - |
|
319 |
+
| 0.1688 | 350 | 0.1559 | - |
|
320 |
+
| 0.1929 | 400 | 0.0614 | - |
|
321 |
+
| 0.2170 | 450 | 0.1498 | - |
|
322 |
+
| 0.2411 | 500 | 0.0587 | - |
|
323 |
+
| 0.2652 | 550 | 0.0278 | - |
|
324 |
+
| 0.2893 | 600 | 0.0741 | - |
|
325 |
+
| 0.3134 | 650 | 0.0271 | - |
|
326 |
+
| 0.3375 | 700 | 0.0404 | - |
|
327 |
+
| 0.3616 | 750 | 0.0399 | - |
|
328 |
+
| 0.3857 | 800 | 0.0152 | - |
|
329 |
+
| 0.4098 | 850 | 0.0186 | - |
|
330 |
+
| 0.4339 | 900 | 0.0365 | - |
|
331 |
+
| 0.4581 | 950 | 0.0175 | - |
|
332 |
+
| 0.4822 | 1000 | 0.0657 | - |
|
333 |
+
| 0.5063 | 1050 | 0.0093 | - |
|
334 |
+
| 0.5304 | 1100 | 0.0203 | - |
|
335 |
+
| 0.5545 | 1150 | 0.0228 | - |
|
336 |
+
| 0.5786 | 1200 | 0.0117 | - |
|
337 |
+
| 0.6027 | 1250 | 0.0193 | - |
|
338 |
+
| 0.6268 | 1300 | 0.016 | - |
|
339 |
+
| 0.6509 | 1350 | 0.0035 | - |
|
340 |
+
| 0.6750 | 1400 | 0.0699 | - |
|
341 |
+
| 0.6991 | 1450 | 0.0334 | - |
|
342 |
+
| 0.7232 | 1500 | 0.0442 | - |
|
343 |
+
| 0.7473 | 1550 | 0.0078 | - |
|
344 |
+
| 0.7715 | 1600 | 0.0114 | - |
|
345 |
+
| 0.7956 | 1650 | 0.0077 | - |
|
346 |
+
| 0.8197 | 1700 | 0.002 | - |
|
347 |
+
| 0.8438 | 1750 | 0.0072 | - |
|
348 |
+
| 0.8679 | 1800 | 0.0459 | - |
|
349 |
+
| 0.8920 | 1850 | 0.0046 | - |
|
350 |
+
| 0.9161 | 1900 | 0.0233 | - |
|
351 |
+
| 0.9402 | 1950 | 0.0076 | - |
|
352 |
+
| 0.9643 | 2000 | 0.0033 | - |
|
353 |
| 0.9884 | 2050 | 0.0036 | - |
|
354 |
|
355 |
### Framework Versions
|
|
|
358 |
- Sentence Transformers: 2.2.2
|
359 |
- Transformers: 4.35.2
|
360 |
- PyTorch: 2.1.0+cu121
|
361 |
+
- Datasets: 2.16.1
|
362 |
- Tokenizers: 0.15.0
|
363 |
|
364 |
## Citation
|
config_setfit.json
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
{
|
2 |
-
"
|
3 |
-
"
|
4 |
}
|
|
|
1 |
{
|
2 |
+
"labels": null,
|
3 |
+
"normalize_embeddings": false
|
4 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 549493968
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ae48f3e712c3db3996f7c51d3e844e4dcc7a60d17f240bbef583294d240deaf
|
3 |
size 549493968
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 475151
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:78d735e135fe6007759e8dfb3fe81cbef3a9a4265a11c691148967cf92fefe02
|
3 |
size 475151
|