Add SetFit model

README.md

@@ -88,11 +88,11 @@ preds = model("I loved the spiderman movie!")
 ## Training Details
 
 ### Training Hyperparameters
-- batch_size: (8, 8)
-- num_epochs: (3, 3)
+- batch_size: (4, 4)
+- num_epochs: (2, 2)
 - max_steps: -1
 - sampling_strategy: oversampling
-- body_learning_rate: (4.843982450599576e-05, 4.843982450599576e-05)
+- body_learning_rate: (1.0036564627932058e-05, 1.0036564627932058e-05)
 - head_learning_rate: 0.01
 - loss: CosineSimilarityLoss
 - distance_metric: cosine_distance
@@ -101,15 +101,15 @@ preds = model("I loved the spiderman movie!")
 - use_amp: False
 - warmup_proportion: 0.1
 - l2_weight: 0.01
-- seed: 36
+- seed: 27
 - eval_max_steps: -1
 - load_best_model_at_end: False
 
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0556 | 1    | 0.157         | -               |
-| 2.7778 | 50   | 0.2482        | -               |
+| 0.0278 | 1    | 0.1869        | -               |
+| 1.3889 | 50   | 0.2376        | -               |
 
 ### Framework Versions
 - Python: 3.11.4

model_head.pkl

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