Push model using huggingface_hub.

README.md

@@ -144,24 +144,28 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
 - num_epochs: (2, 2)
 - max_steps: -1
 - sampling_strategy: oversampling
-- num_iterations: 1
-- body_learning_rate: (0.0003478257195882986, 0.0003478257195882986)
-- head_learning_rate: 0.0003478257195882986
+- num_iterations: 2
+- body_learning_rate: (0.0005201181161511404, 0.0005201181161511404)
+- head_learning_rate: 0.00021200244124154418
 - loss: CosineSimilarityLoss
 - distance_metric: cosine_distance
 - margin: 0.25
 - end_to_end: False
 - use_amp: False
 - warmup_proportion: 0.1
+- max_length: 512
 - seed: 42
 - eval_max_steps: -1
-- load_best_model_at_end: False
+- load_best_model_at_end: True
 
 ### Training Results
-| Epoch  | Step | Training Loss | Validation Loss |
-|:------:|:----:|:-------------:|:---------------:|
-| 0.0909 | 1    | 0.2423        | -               |
+| Epoch   | Step   | Training Loss | Validation Loss |
+|:-------:|:------:|:-------------:|:---------------:|
+| 0.0476  | 1      | 0.2504        | -               |
+| 1.0     | 21     | -             | 0.0691          |
+| **2.0** | **42** | **-**         | **0.0445**      |
 
+* The bold row denotes the saved checkpoint.
 ### Framework Versions
 - Python: 3.10.12
 - SetFit: 1.1.0.dev0

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "sentence-transformers/all-MiniLM-L6-v2",
+  "_name_or_path": "checkpoints/step_42",
   "architectures": [
     "BertModel"
   ],

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:7eace4fcc4d8b481344324ec12237f6f5c298d3185e89dbccf8af6c7ce4eda79
+oid sha256:b6d5bbed3f8e6e3c26eff6f9789b1143287d78fede70452eda4cef3333b833f1
 size 90864192

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:72d4e1bc25dfe4c123670bcacefc119d2d13c6598a487b221c85e8109d6c68b0
+oid sha256:ccc2b30a2521323751b38ef4e29d5ee8e2b6a71e33d25feb0e726b505e7f89dd
 size 44071