Corran commited on
Commit
081f988
1 Parent(s): 2b5939b

Add SetFit model

Browse files
Files changed (5) hide show
  1. 1_Pooling/config.json +1 -1
  2. README.md +42 -42
  3. config_setfit.json +2 -2
  4. model.safetensors +1 -1
  5. model_head.pkl +1 -1
1_Pooling/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "word_embedding_dimension": 512,
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.2609 | - |
313
- | 0.0241 | 50 | 0.2264 | - |
314
- | 0.0482 | 100 | 0.2009 | - |
315
- | 0.0723 | 150 | 0.2175 | - |
316
- | 0.0964 | 200 | 0.1365 | - |
317
- | 0.1205 | 250 | 0.0956 | - |
318
- | 0.1446 | 300 | 0.1399 | - |
319
- | 0.1688 | 350 | 0.1632 | - |
320
- | 0.1929 | 400 | 0.063 | - |
321
- | 0.2170 | 450 | 0.147 | - |
322
- | 0.2411 | 500 | 0.1062 | - |
323
- | 0.2652 | 550 | 0.0369 | - |
324
- | 0.2893 | 600 | 0.0889 | - |
325
- | 0.3134 | 650 | 0.0385 | - |
326
- | 0.3375 | 700 | 0.0636 | - |
327
- | 0.3616 | 750 | 0.0571 | - |
328
- | 0.3857 | 800 | 0.0275 | - |
329
- | 0.4098 | 850 | 0.0231 | - |
330
- | 0.4339 | 900 | 0.049 | - |
331
- | 0.4581 | 950 | 0.0105 | - |
332
- | 0.4822 | 1000 | 0.0478 | - |
333
- | 0.5063 | 1050 | 0.007 | - |
334
- | 0.5304 | 1100 | 0.014 | - |
335
- | 0.5545 | 1150 | 0.0105 | - |
336
- | 0.5786 | 1200 | 0.0095 | - |
337
- | 0.6027 | 1250 | 0.0306 | - |
338
- | 0.6268 | 1300 | 0.012 | - |
339
- | 0.6509 | 1350 | 0.0065 | - |
340
- | 0.6750 | 1400 | 0.0524 | - |
341
- | 0.6991 | 1450 | 0.0546 | - |
342
- | 0.7232 | 1500 | 0.0478 | - |
343
- | 0.7473 | 1550 | 0.0045 | - |
344
- | 0.7715 | 1600 | 0.0084 | - |
345
- | 0.7956 | 1650 | 0.0049 | - |
346
- | 0.8197 | 1700 | 0.0032 | - |
347
- | 0.8438 | 1750 | 0.0118 | - |
348
- | 0.8679 | 1800 | 0.0262 | - |
349
- | 0.8920 | 1850 | 0.003 | - |
350
- | 0.9161 | 1900 | 0.0153 | - |
351
- | 0.9402 | 1950 | 0.0064 | - |
352
- | 0.9643 | 2000 | 0.004 | - |
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.15.0
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
- "normalize_embeddings": false,
3
- "labels": null
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:539b6f74ecd75824e2eb61e08c579df1fd2e3f9d2d636a51a02b24b202928a8c
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:a44b4976c78e5d80c98a3eda590fa5be6530defda6d1c9327f7f6ad88aad7406
3
  size 475151
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78d735e135fe6007759e8dfb3fe81cbef3a9a4265a11c691148967cf92fefe02
3
  size 475151