Add SetFit model

README.md

@@ -1,17 +1,13 @@
 ---
-pipeline_tag: text-classification
+pipeline_tag: sentence-similarity
 tags:
 - sentence-transformers
-- text-classification
-- sentimen-analysis
-library_name: transformers
-language:
-- en
-metrics:
-- accuracy
-- precision
-- recall
+- feature-extraction
+- sentence-similarity
+- transformers
+
 ---
+
 # {MODEL_NAME}
 
 This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 1024 dimensional dense vector space and can be used for tasks like clustering or semantic search.
@@ -91,9 +87,7 @@ The model was trained with the parameters:
 
 `torch.utils.data.dataloader.DataLoader` of length 14756 with parameters:
 ```
-{"multi_target_strategy": "multi-output",
-    "candidate_labels": ['atmosphere',  'beer list',  'bottomless brunch',  'breakfast',  'brunch',  'business',  'byob',  'casual',  'date night',  'delivery',  'fine dining',  'gluten free',  'good value',  'great cocktails',  'great service',  'happy hour',  'healthy',  'instagrammable',  'kid friendly',  'large groups',  'late night',  'live music',  'lunch',  'night out',  'outdoor seating',  'quick',  'romantic',  'rooftop',  'small plates',  'special event',  'sports/tvs',  'takeout',  'tasting menu',  'vegetarian',  'walk-in',  'wine list'],
-   'batch_size': 4, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
+{'batch_size': 4, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
 
 **Loss**:
@@ -103,9 +97,7 @@ The model was trained with the parameters:
 Parameters of the fit()-Method:
 ```
 {
-    "multi_target_strategy": "multi-output",
-    "candidate_labels": ['atmosphere',  'beer list',  'bottomless brunch',  'breakfast',  'brunch',  'business',  'byob',  'casual',  'date night',  'delivery',  'fine dining',  'gluten free',  'good value',  'great cocktails',  'great service',  'happy hour',  'healthy',  'instagrammable',  'kid friendly',  'large groups',  'late night',  'live music',  'lunch',  'night out',  'outdoor seating',  'quick',  'romantic',  'rooftop',  'small plates',  'special event',  'sports/tvs',  'takeout',  'tasting menu',  'vegetarian',  'walk-in',  'wine list'],
-    "epochs": 3,
+    "epochs": 1,
     "evaluation_steps": 0,
     "evaluator": "NoneType",
     "max_grad_norm": 1,

config.json

@@ -1,10 +1,9 @@
 {
-  "_name_or_path": "/root/.cache/torch/sentence_transformers/facebook_bart-large-mnli",
-  "_num_labels": 36,
+  "_name_or_path": "/content/drive/MyDrive/ML-Files/Relabel_Jan_18_2023/temp/models_fewshot_3/",
+  "_num_labels": 3,
   "activation_dropout": 0.0,
   "activation_function": "gelu",
   "add_final_layer_norm": false,
-  "multi_target_strategy": "multi-output",
   "architectures": [
     "BartModel"
   ],
@@ -27,82 +26,16 @@
   "forced_eos_token_id": 2,
   "gradient_checkpointing": false,
   "id2label": {
-    "0": "atmosphere",
-    "1": "beer list",
-    "2": "bottomless brunch",
-    "3": "breakfast",
-    "4": "brunch",
-    "5": "business",
-    "6": "byob",
-    "7": "casual",
-    "8": "date night",
-    "9": "delivery",
-    "10": "fine dining",
-    "11": "gluten free",
-    "12": "good value",
-    "13": "great cocktails",
-    "14": "great service",
-    "15": "happy hour",
-    "16": "healthy",
-    "17": "instagrammable",
-    "18": "kid friendly",
-    "19": "large groups",
-    "20": "late night",
-    "21": "live music",
-    "22": "lunch",
-    "23": "night out",
-    "24": "outdoor seating",
-    "25": "quick",
-    "26": "romantic",
-    "27": "rooftop",
-    "28": "small plates",
-    "29": "special event",
-    "30": "sports/tvs",
-    "31": "takeout",
-    "32": "tasting menu",
-    "33": "vegetarian",
-    "34": "walk-in",
-    "35": "wine list"
+    "0": "contradiction",
+    "1": "neutral",
+    "2": "entailment"
   },
   "init_std": 0.02,
   "is_encoder_decoder": true,
-   "label2id": {
-    "atmosphere": 0,
-    "beer list": 1,
-    "bottomless brunch": 2,
-    "breakfast": 3,
-    "brunch": 4,
-    "business": 5,
-    "byob": 6,
-    "casual": 7,
-    "date night": 8,
-    "delivery": 9,
-    "fine dining": 10,
-    "gluten free": 11,
-    "good value": 12,
-    "great cocktails": 13,
-    "great service": 14,
-    "happy hour": 15,
-    "healthy": 16,
-    "instagrammable": 17,
-    "kid friendly": 18,
-    "large groups": 19,
-    "late night": 20,
-    "live music": 21,
-    "lunch": 22,
-    "night out": 23,
-    "outdoor seating": 24,
-    "quick": 25,
-    "romantic": 26,
-    "rooftop": 27,
-    "small plates": 28,
-    "special event": 29,
-    "sports/tvs": 30,
-    "takeout": 31,
-    "tasting menu": 32,
-    "vegetarian": 33,
-    "walk-in": 34,
-    "wine list": 35
+  "label2id": {
+    "contradiction": 0,
+    "entailment": 2,
+    "neutral": 1
   },
   "max_position_embeddings": 1024,
   "model_type": "bart",
@@ -115,4 +48,4 @@
   "transformers_version": "4.26.0",
   "use_cache": true,
   "vocab_size": 50265
-}
+}

tokenizer_config.json

@@ -6,7 +6,7 @@
   "errors": "replace",
   "mask_token": "<mask>",
   "model_max_length": 1024,
-  "name_or_path": "/root/.cache/torch/sentence_transformers/facebook_bart-large-mnli",
+  "name_or_path": "/content/drive/MyDrive/ML-Files/Relabel_Jan_18_2023/temp/models_fewshot_3/",
   "pad_token": "<pad>",
   "sep_token": "</s>",
   "special_tokens_map_file": null,