Add SetFit model

README.md

@@ -92,7 +92,7 @@ preds = model("I loved the spiderman movie!")
 - num_epochs: (2, 2)
 - max_steps: -1
 - sampling_strategy: oversampling
-- body_learning_rate: (2.260163255777712e-05, 2.260163255777712e-05)
+- body_learning_rate: (0.00036820448519129395, 0.00036820448519129395)
 - 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: 26
+- seed: 17
 - eval_max_steps: -1
 - load_best_model_at_end: False
 
 ### Training Results
 | Epoch | Step | Training Loss | Validation Loss |
 |:-----:|:----:|:-------------:|:---------------:|
-| 0.2   | 1    | 0.2205        | -               |
+| 0.2   | 1    | 0.2562        | -               |
 
 ### 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:f0d3ad8fdc01478a95168c7ea7f8340dfc57285d5ffc3896a19be68961bef7ce
+oid sha256:cac135d3cd547b9df19798a312e09aa1badd5092b6f316069a07782d55ea470a
 size 7067