lucienbaumgartner commited on
Commit
12dc012
·
verified ·
1 Parent(s): 4fc8065

Add SetFit model

Browse files
Files changed (4) hide show
  1. README.md +11 -11
  2. config_setfit.json +4 -4
  3. model.safetensors +1 -1
  4. model_head.pkl +1 -1
README.md CHANGED
@@ -139,19 +139,19 @@ preds = model("Hats off to the design team.")
139
  | Epoch | Step | Training Loss | Validation Loss |
140
  |:-------:|:------:|:-------------:|:---------------:|
141
  | 0.0556 | 1 | 0.3495 | - |
142
- | 1.0 | 18 | - | 0.2237 |
143
- | **2.0** | **36** | **-** | **0.1877** |
144
- | 2.7778 | 50 | 0.0653 | - |
145
- | 3.0 | 54 | - | 0.2112 |
146
- | 4.0 | 72 | - | 0.2118 |
147
  | 5.0 | 90 | - | 0.2202 |
148
  | 5.5556 | 100 | 0.0661 | - |
149
- | 6.0 | 108 | - | 0.2281 |
150
- | 7.0 | 126 | - | 0.23 |
151
- | 8.0 | 144 | - | 0.2355 |
152
- | 8.3333 | 150 | 0.0322 | - |
153
- | 9.0 | 162 | - | 0.2376 |
154
- | 10.0 | 180 | - | 0.2375 |
155
 
156
  * The bold row denotes the saved checkpoint.
157
  ### Framework Versions
 
139
  | Epoch | Step | Training Loss | Validation Loss |
140
  |:-------:|:------:|:-------------:|:---------------:|
141
  | 0.0556 | 1 | 0.3495 | - |
142
+ | 1.0 | 18 | - | 0.2242 |
143
+ | **2.0** | **36** | **-** | **0.1868** |
144
+ | 2.7778 | 50 | 0.0636 | - |
145
+ | 3.0 | 54 | - | 0.2124 |
146
+ | 4.0 | 72 | - | 0.213 |
147
  | 5.0 | 90 | - | 0.2202 |
148
  | 5.5556 | 100 | 0.0661 | - |
149
+ | 6.0 | 108 | - | 0.228 |
150
+ | 7.0 | 126 | - | 0.2306 |
151
+ | 8.0 | 144 | - | 0.2352 |
152
+ | 8.3333 | 150 | 0.0318 | - |
153
+ | 9.0 | 162 | - | 0.2368 |
154
+ | 10.0 | 180 | - | 0.2372 |
155
 
156
  * The bold row denotes the saved checkpoint.
157
  ### Framework Versions
config_setfit.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "normalize_embeddings": false,
3
  "labels": [
4
- "Admiration",
5
  "Gratitude",
6
- "Praise"
7
- ]
 
8
  }
 
1
  {
 
2
  "labels": [
3
+ "Praise",
4
  "Gratitude",
5
+ "Admiration"
6
+ ],
7
+ "normalize_embeddings": false
8
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:981c9c7e8fb68bbe8a84aa32a07de61a83831322f8e4aa0777ad97e71b30c619
3
  size 437967672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5fc33f86dce158903b65d830c5f12cbceb4dc68e1ec3139f2b8d5ac532c2d80
3
  size 437967672
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d22d7f2c00e8bebc33d6b28367d03fdd1284d1fb681394425eafcb6914c566a7
3
  size 20321
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e9c79beaa41bf2546e6d75687d1336f7046f8d9f008ae7b4548dddcaee5a6ac
3
  size 20321