Add SetFit model
Browse files- README.md +16 -14
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
@@ -11,13 +11,15 @@ datasets:
|
|
11 |
metrics:
|
12 |
- accuracy
|
13 |
widget:
|
14 |
-
- text:
|
15 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
- text: LMAO while watching this church burn
|
17 |
-
- text: Islam is a religion of peace.........and we will kill anyone that says otherwise..........in
|
18 |
-
the name of Allah of course!
|
19 |
-
- text: If only there was a way to get children without having to get women pregnant
|
20 |
-
we had them all ready killed
|
21 |
pipeline_tag: text-classification
|
22 |
inference: false
|
23 |
model-index:
|
@@ -32,7 +34,7 @@ model-index:
|
|
32 |
split: test
|
33 |
metrics:
|
34 |
- type: accuracy
|
35 |
-
value: 0.
|
36 |
name: Accuracy
|
37 |
---
|
38 |
|
@@ -68,7 +70,7 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
68 |
### Metrics
|
69 |
| Label | Accuracy |
|
70 |
|:--------|:---------|
|
71 |
-
| **all** | 0.
|
72 |
|
73 |
## Uses
|
74 |
|
@@ -88,7 +90,7 @@ from setfit import SetFitModel
|
|
88 |
# Download from the 🤗 Hub
|
89 |
model = SetFitModel.from_pretrained("amitprgx/setfit-ethos-multilabel-example")
|
90 |
# Run inference
|
91 |
-
preds = model("
|
92 |
```
|
93 |
|
94 |
<!--
|
@@ -120,7 +122,7 @@ preds = model("Waitresses are cunts")
|
|
120 |
### Training Set Metrics
|
121 |
| Training set | Min | Median | Max |
|
122 |
|:-------------|:----|:--------|:----|
|
123 |
-
| Word count |
|
124 |
|
125 |
### Training Hyperparameters
|
126 |
- batch_size: (16, 16)
|
@@ -143,10 +145,10 @@ preds = model("Waitresses are cunts")
|
|
143 |
### Training Results
|
144 |
| Epoch | Step | Training Loss | Validation Loss |
|
145 |
|:------:|:----:|:-------------:|:---------------:|
|
146 |
-
| 0.0063 | 1 | 0.
|
147 |
-
| 0.3125 | 50 | 0.
|
148 |
-
| 0.625 | 100 | 0.
|
149 |
-
| 0.9375 | 150 | 0.
|
150 |
|
151 |
### Framework Versions
|
152 |
- Python: 3.10.12
|
|
|
11 |
metrics:
|
12 |
- accuracy
|
13 |
widget:
|
14 |
+
- text: Are you mentally handicapped or this is your normal form? Both ways kill yourself
|
15 |
+
please.
|
16 |
+
- text: American people are dumbass. They don't even know geography
|
17 |
+
- text: Put a woman to abuse that fat woman and don’t put a kind guy
|
18 |
+
- text: That's what we should do here in a Us survey take them to FEMA camps and put
|
19 |
+
the gas on these people are not coming from starvation destitute bad areas look
|
20 |
+
how they're dressed their well-dressed well fed it's like a trojan horse and it's
|
21 |
+
time we put them in the penal camps and get rid of them make use of the FEMA camp
|
22 |
- text: LMAO while watching this church burn
|
|
|
|
|
|
|
|
|
23 |
pipeline_tag: text-classification
|
24 |
inference: false
|
25 |
model-index:
|
|
|
34 |
split: test
|
35 |
metrics:
|
36 |
- type: accuracy
|
37 |
+
value: 0.44680851063829785
|
38 |
name: Accuracy
|
39 |
---
|
40 |
|
|
|
70 |
### Metrics
|
71 |
| Label | Accuracy |
|
72 |
|:--------|:---------|
|
73 |
+
| **all** | 0.4468 |
|
74 |
|
75 |
## Uses
|
76 |
|
|
|
90 |
# Download from the 🤗 Hub
|
91 |
model = SetFitModel.from_pretrained("amitprgx/setfit-ethos-multilabel-example")
|
92 |
# Run inference
|
93 |
+
preds = model("LMAO while watching this church burn")
|
94 |
```
|
95 |
|
96 |
<!--
|
|
|
122 |
### Training Set Metrics
|
123 |
| Training set | Min | Median | Max |
|
124 |
|:-------------|:----|:--------|:----|
|
125 |
+
| Word count | 3 | 18.1719 | 87 |
|
126 |
|
127 |
### Training Hyperparameters
|
128 |
- batch_size: (16, 16)
|
|
|
145 |
### Training Results
|
146 |
| Epoch | Step | Training Loss | Validation Loss |
|
147 |
|:------:|:----:|:-------------:|:---------------:|
|
148 |
+
| 0.0063 | 1 | 0.3377 | - |
|
149 |
+
| 0.3125 | 50 | 0.1288 | - |
|
150 |
+
| 0.625 | 100 | 0.0798 | - |
|
151 |
+
| 0.9375 | 150 | 0.1517 | - |
|
152 |
|
153 |
### Framework Versions
|
154 |
- Python: 3.10.12
|
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:b6ac0778714040e50528101b03ba629b3352c7c2f2d89b2b9dc4462572524606
|
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 52836
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d2ddaa4e093425e1b6e6789ac250dffe6908931065ff61a6ec7d02efcbf6a1c5
|
3 |
size 52836
|