Add SetFit model

1_Pooling/config.json

@@ -5,5 +5,6 @@
   "pooling_mode_max_tokens": false,
   "pooling_mode_mean_sqrt_len_tokens": false,
   "pooling_mode_weightedmean_tokens": false,
-  "pooling_mode_lasttoken": false
+  "pooling_mode_lasttoken": false,
+  "include_prompt": true
 }

README.md

@@ -8,21 +8,22 @@ tags:
 metrics:
 - accuracy
 widget:
-- text: 'first: We recommend employees start a support group to share and address
-    workplace concerns. second: Grievance Resolution Committee: A committee addresses
-    formal grievances and ensures a fair resolution process.'
-- text: 'first: Supervisors are encouraged to watch TED talks on communication to
-    enhance their skills. second: Progressive Discipline: Disciplinary actions are
-    proportionate and follow a structured process.'
-- text: 'first: Grievance Resolution Committee: A committee addresses formal grievances
-    and ensures a fair resolution process. second: We provide employees with a comprehensive
-    handbook outlining our dispute resolution process for clarity.'
-- text: 'first: We recommend employees seek advice from their peers and mentors to
-    navigate workplace issues. second: We use technology-based solutions to facilitate
-    virtual conflict resolution discussions.'
-- text: 'first: We''ve introduced a complaint of the month contest to highlight and
-    address concerns effectively. second: Our company has a clear conflict resolution
-    policy that all employees must follow.'
+- text: 'first: We recommend self-help books on conflict resolution, available in
+    our office library, as supplemental resources. second: Our company conducts regular
+    surveys to identify and address recurring disputes.'
+- text: 'first: Conflict Resolution Apps: We offer technology solutions for reporting
+    and tracking conflicts. second: Employees can request a mediator to assist in
+    resolving issues with their supervisor, ensuring fair dispute resolution.'
+- text: 'first: Our organization encourages employees to participate in leadership
+    development programs, enhancing their ability to interact with supervisors. second:
+    Conflict Simulation Exercises: Role-playing helps employees practice resolving
+    conflicts.'
+- text: 'first: Mediation sessions are scheduled outside of regular working hours
+    for convenience. second: Employee Conflict Coaches: Coaches work one-on-one with
+    employees to resolve disputes.'
+- text: 'first: We provide conflict resolution pamphlets in the breakroom, offering
+    helpful tips. second: We provide resources for employees to seek external mediation
+    or counseling services if disputes with supervisors persist.'
 pipeline_tag: text-classification
 inference: true
 base_model: sentence-transformers/all-MiniLM-L6-v2
@@ -38,7 +39,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.7272727272727273
+      value: 0.4090909090909091
       name: Accuracy
 ---
 
@@ -80,7 +81,7 @@ The model has been trained using an efficient few-shot learning technique that i
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.7273   |
+| **all** | 0.4091   |
 
 ## Uses
 
@@ -98,9 +99,9 @@ Then you can load this model and run inference.
 from setfit import SetFitModel
 
 # Download from the 🤗 Hub
-model = SetFitModel.from_pretrained("sijan1/setfit-finetuned-fairness")
+model = SetFitModel.from_pretrained("diegofiggie/fairness_model")
 # Run inference
-preds = model("first: We've introduced a complaint of the month contest to highlight and address concerns effectively. second: Our company has a clear conflict resolution policy that all employees must follow.")
+preds = model("first: Mediation sessions are scheduled outside of regular working hours for convenience. second: Employee Conflict Coaches: Coaches work one-on-one with employees to resolve disputes.")
 ```
 
 <!--
@@ -160,44 +161,14 @@ preds = model("first: We've introduced a complaint of the month contest to highl
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.1    | 1    | 0.0141        | -               |
-| 5.0    | 50   | 0.0012        | -               |
-| 10.0   | 100  | 0.0006        | -               |
-| 0.1    | 1    | 0.0005        | -               |
-| 5.0    | 50   | 0.0005        | -               |
-| 10.0   | 100  | 0.0002        | -               |
-| 15.0   | 150  | 0.0002        | -               |
-| 20.0   | 200  | 0.0001        | -               |
-| 25.0   | 250  | 0.0001        | -               |
-| 30.0   | 300  | 0.0001        | -               |
-| 35.0   | 350  | 0.0002        | -               |
-| 40.0   | 400  | 0.0           | -               |
-| 45.0   | 450  | 0.0           | -               |
-| 50.0   | 500  | 0.0           | -               |
-| 55.0   | 550  | 0.0           | -               |
-| 60.0   | 600  | 0.0           | -               |
-| 65.0   | 650  | 0.0001        | -               |
-| 70.0   | 700  | 0.0           | -               |
-| 75.0   | 750  | 0.0           | -               |
-| 80.0   | 800  | 0.0           | -               |
-| 85.0   | 850  | 0.0           | -               |
-| 90.0   | 900  | 0.0           | -               |
-| 95.0   | 950  | 0.0           | -               |
-| 100.0  | 1000 | 0.0           | -               |
-| 0.0667 | 1    | 0.0           | -               |
-| 0.8333 | 50   | 0.0           | -               |
-| 0.0222 | 1    | 0.0           | -               |
-| 0.8333 | 50   | 0.0           | -               |
-| 0.0111 | 1    | 0.0001        | -               |
-| 0.5556 | 50   | 0.0           | -               |
-| 0.0333 | 1    | 0.0           | -               |
+| 0.0333 | 1    | 0.2322        | -               |
 
 ### Framework Versions
-- Python: 3.10.12
+- Python: 3.10.9
 - SetFit: 1.0.3
-- Sentence Transformers: 2.3.1
-- Transformers: 4.37.2
-- PyTorch: 2.1.0+cu121
+- Sentence Transformers: 2.4.0
+- Transformers: 4.38.1
+- PyTorch: 2.2.1+cpu
 - Datasets: 2.17.1
 - Tokenizers: 0.15.2
 

config.json

@@ -19,7 +19,7 @@
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "torch_dtype": "float32",
-  "transformers_version": "4.37.2",
+  "transformers_version": "4.38.1",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

config_sentence_transformers.json

@@ -3,5 +3,7 @@
     "sentence_transformers": "2.0.0",
     "transformers": "4.6.1",
     "pytorch": "1.8.1"
-  }
+  },
+  "prompts": {},
+  "default_prompt_name": null
 }

config_setfit.json

@@ -1,4 +1,4 @@
 {
-  "labels": null,
-  "normalize_embeddings": false
+  "normalize_embeddings": false,
+  "labels": null
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c2ab9e5d8a1781daa63f933ba0a20df6f98fe6a6f6d7802970f3fcc24bfbb8d4
+oid sha256:271165e1148d41d5f71a15902a0759c4515a26e700c5068405f4db54cdcdfde6
 size 90864192

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da534e8c56dac72f1467ca8580929be60367c31d3de44d75ca75ddfc8c941f9c
-size 3935
+oid sha256:2ca976370557f01d234bcfba600462a5c37a6308e8eed07b33f09fe60ca0a181
+size 3813