Add SetFit model

README.md

@@ -88,11 +88,11 @@ preds = model("I loved the spiderman movie!")
 ## Training Details
 
 ### Training Hyperparameters
-- batch_size: (32, 32)
-- num_epochs: (2, 2)
+- batch_size: (16, 16)
+- num_epochs: (3, 3)
 - max_steps: -1
 - sampling_strategy: oversampling
-- body_learning_rate: (0.00036820448519129395, 0.00036820448519129395)
+- body_learning_rate: (1.1001349299683076e-05, 1.1001349299683076e-05)
 - head_learning_rate: 0.01
 - loss: CosineSimilarityLoss
 - distance_metric: cosine_distance
@@ -101,14 +101,14 @@ preds = model("I loved the spiderman movie!")
 - use_amp: False
 - warmup_proportion: 0.1
 - l2_weight: 0.01
-- seed: 17
+- seed: 16
 - eval_max_steps: -1
 - load_best_model_at_end: False
 
 ### Training Results
-| Epoch | Step | Training Loss | Validation Loss |
-|:-----:|:----:|:-------------:|:---------------:|
-| 0.2   | 1    | 0.2562        | -               |
+| Epoch  | Step | Training Loss | Validation Loss |
+|:------:|:----:|:-------------:|:---------------:|
+| 0.1111 | 1    | 0.2312        | -               |
 
 ### Framework Versions
 - Python: 3.11.4

config_setfit.json

@@ -1,7 +1,7 @@
 {
+  "normalize_embeddings": false,
   "labels": [
     "NEUTRAL",
     "TOXIC"
-  ],
-  "normalize_embeddings": false
+  ]
 }

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cac135d3cd547b9df19798a312e09aa1badd5092b6f316069a07782d55ea470a
+oid sha256:dc11d4da7d63298c02bb0be07f85914f3a964da8e393b79a78a209f71c93b105
 size 7067