Add SetFit model

README.md

@@ -141,7 +141,7 @@ preds = model("TikTok creators exploded in popularity.")
 
 ### Training Hyperparameters
 - batch_size: (16, 16)
-- num_epochs: (2, 2)
+- num_epochs: (4, 4)
 - max_steps: -1
 - sampling_strategy: oversampling
 - num_iterations: 20
@@ -162,7 +162,9 @@ preds = model("TikTok creators exploded in popularity.")
 | Epoch | Step | Training Loss | Validation Loss |
 |:-----:|:----:|:-------------:|:---------------:|
 | 0.025 | 1    | 0.0637        | -               |
-| 1.25  | 50   | 0.0322        | -               |
+| 1.25  | 50   | 0.0372        | -               |
+| 2.5   | 100  | 0.0035        | -               |
+| 3.75  | 150  | 0.0017        | -               |
 
 ### Framework Versions
 - Python: 3.10.7

config_setfit.json

@@ -1,4 +1,4 @@
 {
-  "labels": null,
-  "normalize_embeddings": false
+  "normalize_embeddings": false,
+  "labels": null
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b2cb43c7a74a673086d6e4f35c0ade1819b61d7fb3f66f26cde974539b224cc
+oid sha256:52ef91e8bf681f55277b7ea27fc35f9460340a8e544acaed29bfe14c54f67f10
 size 437967672

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f307d67e3c96c04742a9cffec394dc2328dd182504163c5dd9f3792f258f2f06
+oid sha256:9a9db41881decd33047500204c2b8cd4aa39b9d0e84dabb042642125b426c698
 size 50119