Push model using huggingface_hub.

README.md

@@ -141,12 +141,12 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
 
 ### Training Hyperparameters
 - batch_size: (64, 64)
-- num_epochs: (2, 2)
+- num_epochs: (1, 1)
 - max_steps: -1
 - sampling_strategy: oversampling
-- num_iterations: 1
-- body_learning_rate: (0.00014039790878227676, 0.00014039790878227676)
-- head_learning_rate: 0.0005686856164979006
+- num_iterations: 2
+- body_learning_rate: (0.00012410271291092199, 0.00012410271291092199)
+- head_learning_rate: 0.0001551076479757721
 - loss: CosineSimilarityLoss
 - distance_metric: cosine_distance
 - margin: 0.25
@@ -161,9 +161,8 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
 ### Training Results
 | Epoch   | Step   | Training Loss | Validation Loss |
 |:-------:|:------:|:-------------:|:---------------:|
-| 0.0909  | 1      | 0.2491        | -               |
-| 1.0     | 11     | -             | 0.1241          |
-| **2.0** | **22** | **-**         | **0.086**       |
+| 0.0476  | 1      | 0.2484        | -               |
+| **1.0** | **21** | **-**         | **0.1014**      |
 
 * The bold row denotes the saved checkpoint.
 ### Framework Versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "checkpoints/step_22",
+  "_name_or_path": "checkpoints/step_21",
   "architectures": [
     "BertModel"
   ],

config_setfit.json

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:66c49c5c2e94da4d0dc3a52a80b24d4f706279db5dbb4cb2652f1efb37d691a4
+oid sha256:5bbc2bd8d39472e6d55398e4a8426eea02aac2568fa819bf58301211bd62305e
 size 90864192

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0164d0e39c0d0cf8f7597cee8b614192a0356cd5dc4a1b971960c5878a6f2544
+oid sha256:c0b48c9fb9b7381fc9b0fcf3195187185f6e455b747f94651270e9e6d587ebd3
 size 44071