Add SetFit ABSA model

README.md

@@ -6,16 +6,16 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: apps:it remains fast enough for my needs as well, it also has enough storage
-    for my 25gb of music, my entire 20gb of google drive and all my apps, and it runs
-    them all pretty fas
-- text: gb ram:the 16gb soldiered ram performs excellent, and again at this price
-    point, most laptops only have 8gb ram.
-- text: bag:i can carry it around in my backpack and can't tell the difference if
-    i don't have it in my bag.
-- text: opinion:in my opinion, worth the money!
-- text: ram:the ram is good and ssd is good (350mb/s write speed and 1200mb/s read
-    speed).
+- text: fast:fast boot up, great 1080p resolution, expandable (added 4gb additional
+    ram and a 1tb hd) and great value for it's $365+tax price point.
+- text: thinness:the sleekness and thinness of this laptop is lightweight and easy
+    to carry.
+- text: read:when what i heave read, the memory is not upgradeable since it's soldered
+    to the board.
+- text: memory:a good amount of memory. it doesnt need to have a bunch of memory,
+    but a decent amount is perfect!
+- text: wifi:nevertheless great processor, great graphics, 16 gb memory runs cool
+    in daily use, battery lasts about 6-7 hours using wifi and video.
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -61,10 +61,10 @@ This model was trained within the context of a larger system for ABSA, which loo
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
 ### Model Labels
-| Label     | Examples                                                                                                                                                                                                                         |
-|:----------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| aspect    | <ul><li>'memory:painfully slow, you cannot add more memory.'</li><li>'slow:painfully slow, you cannot add more memory.'</li><li>"keyboard:keyboard doesn't light up. hard to see at night"</li></ul>                             |
-| no aspect | <ul><li>"night:keyboard doesn't light up. hard to see at night"</li><li>'chip:the chip, ram, and memory were the selling feature for me.'</li><li>'feature:the chip, ram, and memory were the selling feature for me.'</li></ul> |
+| Label     | Examples                                                                                                                                                                                                                                                                |
+|:----------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| aspect    | <ul><li>'lightweight:very lightweight.'</li><li>'carry:this computer is so light weight and easy to carry.'</li><li>"lightweight:it's lightweight, the screen is decently bright, and it'll go for hours without needing a charge"</li></ul>                            |
+| no aspect | <ul><li>'computer:this computer is so light weight and easy to carry.'</li><li>'weight:this computer is so light weight and easy to carry.'</li><li>"screen:it's lightweight, the screen is decently bright, and it'll go for hours without needing a charge"</li></ul> |
 
 ## Uses
 
@@ -119,12 +119,12 @@ preds = model("The food was great, but the venue is just way too busy.")
 ### Training Set Metrics
 | Training set | Min | Median  | Max |
 |:-------------|:----|:--------|:----|
-| Word count   | 3   | 15.3071 | 29  |
+| Word count   | 1   | 15.9156 | 37  |
 
 | Label     | Training Sample Count |
 |:----------|:----------------------|
-| no aspect | 266                   |
-| aspect    | 154                   |
+| no aspect | 251                   |
+| aspect    | 140                   |
 
 ### Training Hyperparameters
 - batch_size: (128, 128)
@@ -147,17 +147,23 @@ preds = model("The food was great, but the venue is just way too busy.")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0013 | 1    | 0.3011        | -               |
-| 0.0674 | 50   | 0.2947        | 0.2712          |
-| 0.1348 | 100  | 0.2609        | 0.2609          |
-| 0.2022 | 150  | 0.2512        | 0.2532          |
-| 0.2695 | 200  | 0.2383        | 0.2379          |
-| 0.3369 | 250  | 0.1842        | 0.1999          |
-| 0.4043 | 300  | 0.0761        | 0.2341          |
-| 0.4717 | 350  | 0.0279        | 0.2693          |
-| 0.5391 | 400  | 0.0149        | 0.2716          |
-| 0.6065 | 450  | 0.0113        | 0.2782          |
-| 0.6739 | 500  | 0.0121        | 0.2946          |
+| 0.0015 | 1    | 0.29          | -               |
+| 0.0770 | 50   | 0.2977        | 0.2646          |
+| 0.1541 | 100  | 0.2622        | 0.2558          |
+| 0.2311 | 150  | 0.2493        | 0.2482          |
+| 0.3082 | 200  | 0.2347        | 0.2261          |
+| 0.3852 | 250  | 0.1396        | 0.1701          |
+| 0.4622 | 300  | 0.0514        | 0.1434          |
+| 0.5393 | 350  | 0.0227        | 0.1808          |
+| 0.6163 | 400  | 0.0161        | 0.1624          |
+| 0.6934 | 450  | 0.011         | 0.1718          |
+| 0.7704 | 500  | 0.0101        | 0.1731          |
+| 0.8475 | 550  | 0.0089        | 0.1433          |
+| 0.9245 | 600  | 0.0061        | 0.1682          |
+| 1.0015 | 650  | 0.0086        | 0.1627          |
+| 1.0786 | 700  | 0.0078        | 0.1767          |
+| 1.1556 | 750  | 0.0068        | 0.1773          |
+| 1.2327 | 800  | 0.0065        | 0.1766          |
 
 ### Framework Versions
 - Python: 3.11.12

config_setfit.json

@@ -1,9 +1,9 @@
 {
+  "normalize_embeddings": false,
   "labels": [
     "no aspect",
     "aspect"
   ],
   "span_context": 0,
-  "normalize_embeddings": false,
   "spacy_model": "en_core_web_sm"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9816b7bdc227c500735c37da7625d3c8770b2dd04302ce2d8b530f80922ea8f0
+oid sha256:c88bfae907ee6f271b3bd804db18485b1c41496d748b20a3bcee40fa32f14427
 size 90864192

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:85f6ce8e2771f754cf29c55b52744f032d9b9b0209b9dae773ddbfe2c27cdf87
+oid sha256:34bea8c27065e2cbae61d8267fe09f9caff0a9df3d475d7f659a6a2dec1821f3
 size 3919