Add SetFit model

README.md

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

model_head.pkl

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