Add SetFit model

README.md

@@ -7,13 +7,11 @@ tags:
 
 ---
 
-# SetFit model for stance classification from news headlines in Spanish
+# {MODEL_NAME}
 
 This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 384 dimensional dense vector space and can be used for tasks like clustering or semantic search.
 
-sentence-transformers/all-MiniLM-L6-v2 model finetuned on stance classification from news headlines in Spanish.
-
-With three classes: ['deny', 'neutral', 'support']
+<!--- Describe your model here -->
 
 ## Usage (Sentence-Transformers)
 
@@ -48,7 +46,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 60 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 170 with parameters:
 ```
 {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -69,8 +67,8 @@ Parameters of the fit()-Method:
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
-    "steps_per_epoch": 60,
-    "warmup_steps": 6,
+    "steps_per_epoch": 170,
+    "warmup_steps": 17,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5bff2fae433d0397f1b423ee519bb948a9b015d52db3f59706081738ef12513f
-size 10222
+oid sha256:dee6088ffa4615cfcb17e686df7e32288b206d32a41da75c9a0cf28d152110e8
+size 10430

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:df26f62bdb5f7f0d0c2dab9006a86ae8d6737e36fab5982655233e7f7f10c6c8
+oid sha256:251ffa77e98fb7e479e6717cf93b45014f03f57a29fa2f2c8b7a6aa8cac52404
 size 90888945