Add SetFit model

README.md

@@ -202,7 +202,7 @@ preds = model("The HR Learning & Development Specialist designs and delivers wor
 
 ### Training Hyperparameters
 - batch_size: (8, 8)
-- num_epochs: (1, 1)
+- num_epochs: (2, 2)
 - max_steps: -1
 - sampling_strategy: oversampling
 - body_learning_rate: (2e-05, 1e-05)
@@ -222,9 +222,12 @@ preds = model("The HR Learning & Development Specialist designs and delivers wor
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
 | 0.0067 | 1    | 0.1555        | -               |
-| 0.3333 | 50   | 0.1021        | -               |
-| 0.6667 | 100  | 0.0053        | -               |
-| 1.0    | 150  | 0.0012        | 0.0723          |
+| 0.3333 | 50   | 0.1102        | -               |
+| 0.6667 | 100  | 0.0045        | -               |
+| 1.0    | 150  | 0.001         | 0.0750          |
+| 1.3333 | 200  | 0.0005        | -               |
+| 1.6667 | 250  | 0.0004        | -               |
+| 2.0    | 300  | 0.0003        | 0.0787          |
 
 ### Framework Versions
 - Python: 3.12.8

config_setfit.json

@@ -1,9 +1,9 @@
 {
+  "normalize_embeddings": false,
   "labels": [
     "IRC",
     "EIC",
     "ESC",
     "SEC"
-  ],
-  "normalize_embeddings": false
+  ]
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:986ab33945f7099b3368642b49761ab78c5918fd798bdc1347b8bc3c24be3fcc
+oid sha256:72b96f4302c5628ad68f63a5d29824d72fd5b9a6e84f1f76fbadd1d6b2d537aa
 size 437967672

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:85336517772ef91c1ea39583a84d74875417323693b94701011ef5b07dedb1f7
+oid sha256:6e9c81188d19d280f6f34263ca3116b297151b41ecb15dfda09b724a9af467ff
 size 25495