Add SetFit ABSA model
Browse files- README.md +19 -20
- config_setfit.json +1 -1
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
@@ -6,14 +6,14 @@ tags:
|
|
6 |
- text-classification
|
7 |
- generated_from_setfit_trainer
|
8 |
widget:
|
9 |
-
- text:
|
10 |
-
|
11 |
-
- text:
|
12 |
-
|
13 |
-
- text:
|
14 |
-
|
15 |
-
- text:
|
16 |
-
|
17 |
metrics:
|
18 |
- accuracy
|
19 |
pipeline_tag: text-classification
|
@@ -59,10 +59,10 @@ This model was trained within the context of a larger system for ABSA, which loo
|
|
59 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
60 |
|
61 |
### Model Labels
|
62 |
-
| Label | Examples
|
63 |
-
|
64 |
-
|
|
65 |
-
|
|
66 |
|
67 |
## Uses
|
68 |
|
@@ -117,12 +117,12 @@ preds = model("The food was great, but the venue is just way too busy.")
|
|
117 |
### Training Set Metrics
|
118 |
| Training set | Min | Median | Max |
|
119 |
|:-------------|:----|:--------|:----|
|
120 |
-
| Word count | 5 | 13.
|
121 |
|
122 |
| Label | Training Sample Count |
|
123 |
|:---------|:----------------------|
|
124 |
-
| negative |
|
125 |
-
| positive |
|
126 |
|
127 |
### Training Hyperparameters
|
128 |
- batch_size: (128, 128)
|
@@ -145,11 +145,10 @@ preds = model("The food was great, but the venue is just way too busy.")
|
|
145 |
### Training Results
|
146 |
| Epoch | Step | Training Loss | Validation Loss |
|
147 |
|:------:|:----:|:-------------:|:---------------:|
|
148 |
-
| 0.
|
149 |
-
| 1.
|
150 |
-
| 2.
|
151 |
-
|
|
152 |
-
| 4.6512 | 200 | 0.0002 | 0.0001 |
|
153 |
|
154 |
### Framework Versions
|
155 |
- Python: 3.11.12
|
|
|
6 |
- text-classification
|
7 |
- generated_from_setfit_trainer
|
8 |
widget:
|
9 |
+
- text: picture quality, easy to use.:loads of features, great picture quality, easy
|
10 |
+
to use.
|
11 |
+
- text: very good auto focus, and a:has a timer that keeps its setting for multiple
|
12 |
+
pictures, has very good auto focus, and a large view screen.
|
13 |
+
- text: takes horrible photos and not easy:takes horrible photos and not easy to use.
|
14 |
+
- text: pictures are very clear and precise.:the pictures are very clear and precise.
|
15 |
+
- text: compact and the images are sharp and:it is compact and the images are sharp
|
16 |
+
and clear.
|
17 |
metrics:
|
18 |
- accuracy
|
19 |
pipeline_tag: text-classification
|
|
|
59 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
60 |
|
61 |
### Model Labels
|
62 |
+
| Label | Examples |
|
63 |
+
|:---------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
64 |
+
| positive | <ul><li>', auto focus excellent:good starter, easy use, auto focus excellent'</li><li>'zoom works very well:zoom works very well.'</li><li>'auto focus is great and:auto focus is great and take superb close ups.'</li></ul> |
|
65 |
+
| negative | <ul><li>"lens that it comes:lens that it comes with definitely ain't for long distances, and the picture quality isn't the best."</li><li>'the zoom on the camera:the zoom on the camera seemed to have a problem, too.'</li><li>'focus is slow,:focus is slow, no continuous options, no touch screen like my older camera.'</li></ul> |
|
66 |
|
67 |
## Uses
|
68 |
|
|
|
117 |
### Training Set Metrics
|
118 |
| Training set | Min | Median | Max |
|
119 |
|:-------------|:----|:--------|:----|
|
120 |
+
| Word count | 5 | 13.8511 | 31 |
|
121 |
|
122 |
| Label | Training Sample Count |
|
123 |
|:---------|:----------------------|
|
124 |
+
| negative | 47 |
|
125 |
+
| positive | 47 |
|
126 |
|
127 |
### Training Hyperparameters
|
128 |
- batch_size: (128, 128)
|
|
|
145 |
### Training Results
|
146 |
| Epoch | Step | Training Loss | Validation Loss |
|
147 |
|:------:|:----:|:-------------:|:---------------:|
|
148 |
+
| 0.0278 | 1 | 0.3158 | - |
|
149 |
+
| 1.3889 | 50 | 0.0839 | 0.0013 |
|
150 |
+
| 2.7778 | 100 | 0.0004 | 0.0006 |
|
151 |
+
| 4.1667 | 150 | 0.0002 | 0.0005 |
|
|
|
152 |
|
153 |
### Framework Versions
|
154 |
- Python: 3.11.12
|
config_setfit.json
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
{
|
|
|
2 |
"labels": [
|
3 |
"negative",
|
4 |
"positive"
|
5 |
],
|
6 |
"span_context": 3,
|
7 |
-
"normalize_embeddings": false,
|
8 |
"spacy_model": "en_core_web_sm"
|
9 |
}
|
|
|
1 |
{
|
2 |
+
"normalize_embeddings": false,
|
3 |
"labels": [
|
4 |
"negative",
|
5 |
"positive"
|
6 |
],
|
7 |
"span_context": 3,
|
|
|
8 |
"spacy_model": "en_core_web_sm"
|
9 |
}
|
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:d4de874438f44f92a7e655ee31606c85290afa6e3da1a3a83b4f0f35cf5ea095
|
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 7055
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2d1bdc66d165b33f1b4e395e179d385a3d8c287de4518a00302072cdd93d6f7f
|
3 |
size 7055
|