Add SetFit model
Browse files- README.md +11 -11
- config_setfit.json +4 -4
- model.safetensors +1 -1
- 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.
|
143 |
-
| **2.0** | **36** | **-** | **0.
|
144 |
-
| 2.7778 | 50 | 0.
|
145 |
-
| 3.0 | 54 | - | 0.
|
146 |
-
| 4.0 | 72 | - | 0.
|
147 |
| 5.0 | 90 | - | 0.2202 |
|
148 |
| 5.5556 | 100 | 0.0661 | - |
|
149 |
-
| 6.0 | 108 | - | 0.
|
150 |
-
| 7.0 | 126 | - | 0.
|
151 |
-
| 8.0 | 144 | - | 0.
|
152 |
-
| 8.3333 | 150 | 0.
|
153 |
-
| 9.0 | 162 | - | 0.
|
154 |
-
| 10.0 | 180 | - | 0.
|
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 |
-
"
|
5 |
"Gratitude",
|
6 |
-
"
|
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:
|
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:
|
3 |
size 20321
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5e9c79beaa41bf2546e6d75687d1336f7046f8d9f008ae7b4548dddcaee5a6ac
|
3 |
size 20321
|