najwaa commited on
Commit
ae07ffb
·
verified ·
1 Parent(s): 9b1b73f

Add SetFit ABSA model

Browse files
Files changed (4) hide show
  1. README.md +23 -26
  2. config_setfit.json +1 -1
  3. model.safetensors +1 -1
  4. 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: out because the screen was broken initially:never had a chance to try it out
10
- because the screen was broken initially.
11
- - text: ' the backlit keyboard is a lie: the backlit keyboard is a lie.'
12
- - text: 118 gb of storage and 4 gb:without the added drive, 118 gb of storage and
13
- 4 gb of ram is more than adequate for basic computing
14
- - text: 4gb of ram and 64gb:4gb of ram and 64gb of storage is not enough for windows
15
- 10.
16
- - text: likely not enough memory or something seems:likely not enough memory or something
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
- | negative | <ul><li>'not add more memory.:painfully slow, you cannot add more memory.'</li><li>'painfully slow, you can:painfully slow, you cannot add more memory.'</li><li>"keyboard doesn't light:keyboard doesn't light up. hard to see at night"</li></ul> |
66
- | positive | <ul><li>'the chip, ram, and memory:the chip, ram, and memory were the selling feature for me.'</li><li>'ram, and memory were the selling:the chip, ram, and memory were the selling feature for me.'</li><li>'very easy to carry.:works fast and very easy to carry.'</li></ul> |
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 | 5 | 16.1364 | 35 |
122
 
123
  | Label | Training Sample Count |
124
  |:---------|:----------------------|
125
- | negative | 73 |
126
- | positive | 81 |
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.0105 | 1 | 0.2887 | - |
150
- | 0.5263 | 50 | 0.1404 | 0.0018 |
151
- | 1.0526 | 100 | 0.0007 | 0.0004 |
152
- | 1.5789 | 150 | 0.0002 | 0.0003 |
153
- | 2.1053 | 200 | 0.0002 | 0.0003 |
154
- | 2.6316 | 250 | 0.0001 | 0.0002 |
155
- | 3.1579 | 300 | 0.0001 | 0.0002 |
156
- | 3.6842 | 350 | 0.0001 | 0.0002 |
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:485c8f3342d25419800431924b3a0e6956451aa28398a84a08856d0bf90a572a
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:40e97e2e280ec2a0930d6bb61da89ffcab46c0e831db7a3f2254b58cff2f8178
3
  size 7055
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:835fb1477be04db03d83d4c6b041ff071531e9e1f3b7cd65320d49b96e033d49
3
  size 7055