Add SetFit ABSA model

README.md

@@ -10,22 +10,20 @@ base_model: BAAI/bge-small-en-v1.5
 metrics:
 - accuracy
 widget:
-- text: to be a nightmare, said retired:The Type 096s are going to be a nightmare,
-    said retired submariner and naval technical intelligence analyst Christopher Carlson,
-    one of the researchers.
-- text: census as an independent exercise.:In fact, the government of Bihar has recently
-    taken up the caste census as an independent exercise.
-- text: to Moscow's Improved Akula boats.:Carlson told Reuters he did not believe
-    China had obtained Russia's 'crown jewels' - its very latest technology - but
-    would be producing a submarine stealthy enough to compare to Moscow's Improved
-    Akula boats.
-- text: staging fully armed nuclear deterrence patrols with its older:The Chinese
-    navy is routinely staging fully armed nuclear deterrence patrols with its older
-    Type 094 boats out of Hainan Island in the South China Sea, the Pentagon said
-    in November, much like patrols operated for years by the United States, Britain,
-    Russia, and France.
-- text: Sanjeev Chopra is a former:Sanjeev Chopra is a former IAS officer and Festival
-    Director of Valley of Words.
+- text: mine employment is starker at mid-:The future of coal mine employment is starker
+    at mid-century.
+- text: are considered relatively noisy - a major:But the Type 094s, which carry China's
+    most advanced submarine-launched JL-3 missile, are considered relatively noisy
+    - a major handicap for military submarines.
+- text: March this year raised an objection, saying that:Initially, the registrar's
+    office of the IHC in March this year raised an objection, saying that the high
+    court was not an appropriate forum and asked the petitioner to approach the relevant
+    authorities.
+- text: mine closures is West Bengal.:Runa Sarkar, a professor at the Indian Institute
+    of Management Calcutta, said the coal mining region most affected by mine closures
+    is West Bengal.
+- text: FIA DG to proceed in accordance with the law.:When the petition was heard
+    by the chief justice, he asked the FIA DG to proceed in accordance with the law.
 pipeline_tag: text-classification
 inference: false
 model-index:
@@ -40,7 +38,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.8387096774193549
+      value: 0.7065217391304348
       name: Accuracy
 ---
 
@@ -93,7 +91,7 @@ This model was trained within the context of a larger system for ABSA, which loo
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.8387   |
+| **all** | 0.7065   |
 
 ## Uses
 
@@ -177,11 +175,11 @@ preds = model("The food was great, but the venue is just way too busy.")
 ### Training Results
 | Epoch      | Step   | Training Loss | Validation Loss |
 |:----------:|:------:|:-------------:|:---------------:|
-| 0.0217     | 1      | 0.2604        | -               |
-| **1.0870** | **50** | **0.0642**    | **0.3479**      |
-| 2.1739     | 100    | 0.0249        | 0.3974          |
-| 3.2609     | 150    | 0.0146        | 0.4341          |
-| 4.3478     | 200    | 0.0033        | 0.4358          |
+| 0.0217     | 1      | 0.2599        | -               |
+| **1.0870** | **50** | **0.0608**    | **0.3526**      |
+| 2.1739     | 100    | 0.0253        | 0.4091          |
+| 3.2609     | 150    | 0.0159        | 0.4497          |
+| 4.3478     | 200    | 0.0035        | 0.4437          |
 
 * The bold row denotes the saved checkpoint.
 ### Framework Versions

config_setfit.json

@@ -5,7 +5,7 @@
     "Negative",
     "Positive"
   ],
-  "spacy_model": "en_core_web_lg",
   "normalize_embeddings": false,
-  "span_context": 3
+  "span_context": 3,
+  "spacy_model": "en_core_web_lg"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d516dab61975def5ed7af94623f6e76b0cf8eb7b8332ef31a6ea942eed896cc9
+oid sha256:8b50fb1bc040ac672ac71f796586a6d07bc3d9f0d110083cc4af858de9828b5b
 size 133462128

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:904885f87a2ed02f4cbd2bd8e8bca759d569cc80af225a44d28a852dea535c42
+oid sha256:d3f74a1b2f81d58e52070703c6970427dbf0197a5248fafb502dc9bf36046414
 size 13335