Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text
+training_progress.png filter=lfs diff=lfs merge=lfs -text

best_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e7e4110b546e050a06196e155ce4617538ea38f293bec307cba5642054ee902
+size 1114896017

config.json

@@ -0,0 +1,25 @@
+{
+  "model_name": "intfloat/multilingual-e5-base",
+  "hidden_size": 768,
+  "num_age_groups": 6,
+  "age_groups": [
+    "1-2",
+    "12-14",
+    "15-19",
+    "3-5",
+    "6-11",
+    "general"
+  ],
+  "age_to_id": {
+    "1-2": 0,
+    "12-14": 1,
+    "15-19": 2,
+    "3-5": 3,
+    "6-11": 4,
+    "general": 5
+  },
+  "max_length": 512,
+  "best_val_loss": 0.15584548979016546,
+  "num_samples": 5228,
+  "final_temperature": 0.0668264776468277
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1eb8d290232d597d8db49bf85fac72c41a4f0373f88185a41a9931c2cb4229d2
+size 1114896662

sentencepiece.bpe.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
+size 5069051

special_tokens_map.json

@@ -0,0 +1,51 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "cls_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a56def25aa40facc030ea8b0b87f3688e4b3c39eb8b45d5702b3a1300fe2a20
+size 17082734

tokenizer_config.json

@@ -0,0 +1,55 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "250001": {
+      "content": "<mask>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "extra_special_tokens": {},
+  "mask_token": "<mask>",
+  "model_max_length": 512,
+  "pad_token": "<pad>",
+  "sep_token": "</s>",
+  "tokenizer_class": "XLMRobertaTokenizer",
+  "unk_token": "<unk>"
+}

training_history.json

@@ -0,0 +1,42 @@
+{
+  "train_loss": [
+    0.30645247583895696,
+    0.13542856172036774,
+    0.07764055220798873,
+    0.04222178316161552,
+    0.0254023968186441,
+    0.007971447666783428
+  ],
+  "val_loss": [
+    0.17906199747924018,
+    0.15584548979016546,
+    0.15711626242920676,
+    0.19052041062264793,
+    0.19481491100747245,
+    0.26780188390916543
+  ],
+  "train_margin": [
+    0.15745257120986172,
+    0.3756912477542219,
+    0.4690174054370697,
+    0.5174879973762421,
+    0.5503568833441713,
+    0.5749395069987493
+  ],
+  "val_margin": [
+    0.34743018846475443,
+    0.4853911267892095,
+    0.5323189945166348,
+    0.5677667312494671,
+    0.5933685687207083,
+    0.6074129097334301
+  ],
+  "temperature": [
+    0.06878325343132019,
+    0.0668264776468277,
+    0.06354578584432602,
+    0.05926638841629028,
+    0.05477049946784973,
+    0.05033325031399727
+  ]
+}