Add SetFit ABSA model
Browse files- README.md +23 -26
- config_setfit.json +1 -1
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
@@ -6,15 +6,14 @@ tags:
|
|
6 |
- text-classification
|
7 |
- generated_from_setfit_trainer
|
8 |
widget:
|
9 |
-
- text:
|
10 |
-
|
11 |
-
|
12 |
-
- text:
|
13 |
-
|
14 |
-
- text:
|
15 |
-
|
16 |
-
|
17 |
-
seems to me
|
18 |
metrics:
|
19 |
- accuracy
|
20 |
pipeline_tag: text-classification
|
@@ -60,10 +59,10 @@ This model was trained within the context of a larger system for ABSA, which loo
|
|
60 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
61 |
|
62 |
### Model Labels
|
63 |
-
| Label | Examples
|
64 |
-
|
65 |
-
|
|
66 |
-
|
|
67 |
|
68 |
## Uses
|
69 |
|
@@ -118,12 +117,12 @@ preds = model("The food was great, but the venue is just way too busy.")
|
|
118 |
### Training Set Metrics
|
119 |
| Training set | Min | Median | Max |
|
120 |
|:-------------|:----|:--------|:----|
|
121 |
-
| Word count |
|
122 |
|
123 |
| Label | Training Sample Count |
|
124 |
|:---------|:----------------------|
|
125 |
-
| negative |
|
126 |
-
| positive |
|
127 |
|
128 |
### Training Hyperparameters
|
129 |
- batch_size: (128, 128)
|
@@ -146,16 +145,14 @@ preds = model("The food was great, but the venue is just way too busy.")
|
|
146 |
### Training Results
|
147 |
| Epoch | Step | Training Loss | Validation Loss |
|
148 |
|:------:|:----:|:-------------:|:---------------:|
|
149 |
-
| 0.
|
150 |
-
| 0.
|
151 |
-
| 1.
|
152 |
-
| 1.
|
153 |
-
| 2.
|
154 |
-
|
|
155 |
-
| 3.
|
156 |
-
|
|
157 |
-
| 4.2105 | 400 | 0.0001 | 0.0002 |
|
158 |
-
| 4.7368 | 450 | 0.0001 | 0.0002 |
|
159 |
|
160 |
### Framework Versions
|
161 |
- Python: 3.11.12
|
|
|
6 |
- text-classification
|
7 |
- generated_from_setfit_trainer
|
8 |
widget:
|
9 |
+
- text: has excellent battery life.:has excellent battery life.
|
10 |
+
- text: '...then the screen went black &:was working fine although we didn''t really
|
11 |
+
use it that much...then the screen went black & nothing.'
|
12 |
+
- text: is easy to carry, this is:other than this, if you need a basic laptop that
|
13 |
+
surfs the internet and is easy to carry, this is for you.
|
14 |
+
- text: ' the price:excellent fot the price'
|
15 |
+
- text: it is fast and performs all:it is fast and performs all the basic functions
|
16 |
+
i need quickly.
|
|
|
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>'very lightweight.:very lightweight.'</li><li>'and easy to carry.:this computer is so light weight and easy to carry.'</li><li>"it's lightweight, the screen:it's lightweight, the screen is decently bright, and it'll go for hours without needing a charge"</li></ul> |
|
65 |
+
| negative | <ul><li>'me a black screen.:defective...wouldnt boot up, just gives me a black screen.'</li><li>'screen often blanks momentarily:screen often blanks momentarily and internet connection is spotty.'</li><li>'it but the screen stopped working after:loved it but the screen stopped working after 4 months'</li></ul> |
|
66 |
|
67 |
## Uses
|
68 |
|
|
|
117 |
### Training Set Metrics
|
118 |
| Training set | Min | Median | Max |
|
119 |
|:-------------|:----|:--------|:----|
|
120 |
+
| Word count | 1 | 16.7714 | 42 |
|
121 |
|
122 |
| Label | Training Sample Count |
|
123 |
|:---------|:----------------------|
|
124 |
+
| negative | 64 |
|
125 |
+
| positive | 76 |
|
126 |
|
127 |
### Training Hyperparameters
|
128 |
- batch_size: (128, 128)
|
|
|
145 |
### Training Results
|
146 |
| Epoch | Step | Training Loss | Validation Loss |
|
147 |
|:------:|:----:|:-------------:|:---------------:|
|
148 |
+
| 0.0127 | 1 | 0.3141 | - |
|
149 |
+
| 0.6329 | 50 | 0.1338 | 0.0015 |
|
150 |
+
| 1.2658 | 100 | 0.0006 | 0.0003 |
|
151 |
+
| 1.8987 | 150 | 0.0002 | 0.0003 |
|
152 |
+
| 2.5316 | 200 | 0.0002 | 0.0002 |
|
153 |
+
| 3.1646 | 250 | 0.0001 | 0.0002 |
|
154 |
+
| 3.7975 | 300 | 0.0001 | 0.0002 |
|
155 |
+
| 4.4304 | 350 | 0.0001 | 0.0002 |
|
|
|
|
|
156 |
|
157 |
### Framework Versions
|
158 |
- 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:a2b8de9c7e208a3a7522dc9d1e4290fda79fd3d4ebb81981aeeb73215dbd7044
|
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:835fb1477be04db03d83d4c6b041ff071531e9e1f3b7cd65320d49b96e033d49
|
3 |
size 7055
|