Add SetFit ABSA model

README.md

@@ -10,22 +10,22 @@ base_model: BAAI/bge-small-en-v1.5
 metrics:
 - accuracy
 widget:
-- text: People:Based partly on Chinese military journals, internal speeches by senior
-    People's Liberation Army (PLA) officers, and patent data, the paper charts more
-    than 50 years of the PLA navy's often-glacial nuclear submarine development.
-- text: Qingdao:Chinese Navy's nuclear-powered submarine Long March 11 takes part
-    in a naval parade off the eastern port city of Qingdao to mark the 70th anniversary
-    of the founding of the Chinese People's Liberation Army Navy.
-- text: warfare drills:Anti-submarine warfare drills are increasing, as are deployments
-    of sub-hunting P-8 Poseidon aircraft around Southeast Asia and the Indian Ocean.
-- text: devices:The research also details potential breakthroughs in specific areas,
-    including pump-jet propulsion and internal quieting devices, based on 'imitative
-    innovation' of Russian technology.
-- text: axe 73,800 jobs:State-run miner Coal India Limited (CIL), which has the biggest
-    workforce among listed government undertakings, is likely to axe 73,800 jobs by
-    2050 as India pledges to move from fossil fuels to green power, according to a
-    research report released by the US-based think tank Global Energy Monitor (GEM)
-    on October 10.
+- text: closures: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: comment:Neither the Russian nor the Chinese defence ministries responded to
+    Reuters' requests for comment.
+- text: 'Canada:The statements made in Canada''s parliament were finally an acknowledgement
+    of the reality that young Sikhs like me have lived through for decades: Sikh dissidents
+    expressing their support for an independent state may face the risk of imminent
+    harm, even in the diaspora.'
+- text: France:The Paris Agreement, a legally binding international treaty on climate
+    change adopted by 196 parties at the UN Climate Change Conference (COP21) in Paris,
+    France in December 2015, aims to hold the increase in the global average temperature
+    to well below 2°C above pre-industrial levels.
+- text: 'risk:The statements made in Canada''s parliament were finally an acknowledgement
+    of the reality that young Sikhs like me have lived through for decades: Sikh dissidents
+    expressing their support for an independent state may face the risk of imminent
+    harm, even in the diaspora.'
 pipeline_tag: text-classification
 inference: false
 model-index:
@@ -40,7 +40,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.7874720357941835
+      value: 0.7630057803468208
       name: Accuracy
 ---
 
@@ -91,7 +91,7 @@ This model was trained within the context of a larger system for ABSA, which loo
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.7875   |
+| **all** | 0.7630   |
 
 ## Uses
 
@@ -174,14 +174,15 @@ preds = model("The food was great, but the venue is just way too busy.")
 | Epoch      | Step    | Training Loss | Validation Loss |
 |:----------:|:-------:|:-------------:|:---------------:|
 | 0.0018     | 1       | 0.2598        | -               |
-| 0.0893     | 50      | 0.2458        | 0.2552          |
-| 0.1786     | 100     | 0.2418        | 0.2527          |
-| **0.2679** | **150** | **0.2427**    | **0.2459**      |
-| 0.3571     | 200     | 0.1272        | 0.2566          |
-| 0.4464     | 250     | 0.0075        | 0.3028          |
-| 0.5357     | 300     | 0.0023        | 0.3251          |
-| 0.625      | 350     | 0.0021        | 0.328           |
-| 0.7143     | 400     | 0.0037        | 0.355           |
+| 0.0893     | 50      | 0.2458        | 0.2547          |
+| 0.1786     | 100     | 0.2418        | 0.2522          |
+| 0.2679     | 150     | 0.2427        | 0.2452          |
+| **0.3571** | **200** | **0.1272**    | **0.2419**      |
+| 0.4464     | 250     | 0.0075        | 0.2853          |
+| 0.5357     | 300     | 0.0023        | 0.3134          |
+| 0.625      | 350     | 0.0021        | 0.3138          |
+| 0.7143     | 400     | 0.0037        | 0.3502          |
+| 0.8036     | 450     | 0.011         | 0.3437          |
 
 * The bold row denotes the saved checkpoint.
 ### Framework Versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "models/step_150",
+  "_name_or_path": "models/step_200",
   "architectures": [
     "BertModel"
   ],

config_setfit.json

@@ -3,7 +3,7 @@
     "no aspect",
     "aspect"
   ],
-  "spacy_model": "en_core_web_lg",
   "normalize_embeddings": false,
-  "span_context": 0
+  "span_context": 0,
+  "spacy_model": "en_core_web_lg"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:632ad8463ed94b5d6918d13be6dc388d1cfe583a2fe3d469f84c6312a9855aca
+oid sha256:f561bf1872dee4ee28bf0dce9928b671ec79f80f4d59898c3c89d4d9c3e1220a
 size 133462128

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:376562608e572c6d4e3f76ac297b53f1c85ce31b41fc9e965a607eb8394dfb3d
+oid sha256:361558fdb7f6f2124a2dca2ef02494aa04c0358f23442e8a804dc898c16b30df
 size 3919