Upload ONNX router (logits+variance) + thresholds + label map + tokenizer

Browse files

Files changed (10) hide show

README.md +7 -0
label_map.json +24 -0
metrics.json +12 -0
model.onnx +3 -0
router_meta.json +13 -0
special_tokens_map.json +37 -0
thresholds.json +30 -0
tokenizer.json +0 -0
tokenizer_config.json +65 -0
vocab.txt +0 -0

README.md ADDED Viewed

	@@ -0,0 +1,7 @@

+# lef-neural-controller
+Artifacts:
+- model.onnx (outputs: logits, variance)
+- thresholds.json
+- label_map.json
+- tokenizer files (if present)

label_map.json ADDED Viewed

	@@ -0,0 +1,24 @@

+{
+  "intent2id": {
+    "INTENT_AUTH_OK": 0,
+    "INTENT_AUTH_NOK": 1,
+    "INTENT_GREETING": 2,
+    "INTENT_PRODUCT_FAQ": 3,
+    "INTENT_INTEREST_DIGITAL": 4,
+    "INTENT_INTEREST_PHYSICAL": 5,
+    "INTENT_NO_INTEREST": 6,
+    "INTENT_CONFIRM_APPOINTMENT_PHYSICAL": 7,
+    "INTENT_CONFIRM_OBTENTION_DIGITAL": 8
+  },
+  "id2intent": {
+    "0": "INTENT_AUTH_OK",
+    "1": "INTENT_AUTH_NOK",
+    "2": "INTENT_GREETING",
+    "3": "INTENT_PRODUCT_FAQ",
+    "4": "INTENT_INTEREST_DIGITAL",
+    "5": "INTENT_INTEREST_PHYSICAL",
+    "6": "INTENT_NO_INTEREST",
+    "7": "INTENT_CONFIRM_APPOINTMENT_PHYSICAL",
+    "8": "INTENT_CONFIRM_OBTENTION_DIGITAL"
+  }
+}

metrics.json ADDED Viewed

	@@ -0,0 +1,12 @@

+{
+  "ood_auroc_var": 0.7421442470654698,
+  "coverage_in_domain": 0.8296089385474861,
+  "selective_risk_in_domain": 0.006734006734006734,
+  "ece_in_domain_accepted": 0.07366302609443665,
+  "abstain_rate_calculated_all": 0.45828144458281445,
+  "abstain_rate_ood_all": 0.1133250311332503,
+  "ort_fp32_avg_ms_batch1": 29.16112204000001,
+  "device_train": "cuda",
+  "model_name": "sentence-transformers/all-MiniLM-L6-v2",
+  "max_len": 64
+}

model.onnx ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:6e5de48085a86235c011a5cea95eca0f7f788081805447f043a65350b217044e
+size 96804779

router_meta.json ADDED Viewed

	@@ -0,0 +1,13 @@

+{
+  "format": "onnxruntime",
+  "onnx_file": "model.onnx",
+  "inputs": [
+    "input_ids",
+    "attention_mask"
+  ],
+  "outputs": [
+    "logits",
+    "variance"
+  ],
+  "max_length": 64
+}

special_tokens_map.json ADDED Viewed

	@@ -0,0 +1,37 @@

+{
+  "cls_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "[MASK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "[PAD]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "[UNK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

thresholds.json ADDED Viewed

	@@ -0,0 +1,30 @@

+{
+  "version": "sngp_crc_v1",
+  "tau_ood": 3.2301740646362305,
+  "mean_field_c": 0.39269908169872414,
+  "lambda_by_intent": {
+    "INTENT_AUTH_OK": 0.5496541857719421,
+    "INTENT_AUTH_NOK": 0.8975508213043213,
+    "INTENT_GREETING": 0.2773037552833557,
+    "INTENT_PRODUCT_FAQ": 0.24596728384494781,
+    "INTENT_INTEREST_DIGITAL": 0.8745520710945129,
+    "INTENT_INTEREST_PHYSICAL": 0.7114467024803162,
+    "INTENT_NO_INTEREST": 0.48147109150886536,
+    "INTENT_CONFIRM_APPOINTMENT_PHYSICAL": 0.3335312604904175,
+    "INTENT_CONFIRM_OBTENTION_DIGITAL": 0.5463740825653076
+  },
+  "alpha_by_intent": {
+    "INTENT_AUTH_OK": 0.03,
+    "INTENT_AUTH_NOK": 0.03,
+    "INTENT_GREETING": 0.1,
+    "INTENT_PRODUCT_FAQ": 0.1,
+    "INTENT_INTEREST_DIGITAL": 0.03,
+    "INTENT_INTEREST_PHYSICAL": 0.03,
+    "INTENT_NO_INTEREST": 0.05,
+    "INTENT_CONFIRM_APPOINTMENT_PHYSICAL": 0.05,
+    "INTENT_CONFIRM_OBTENTION_DIGITAL": 0.05
+  },
+  "calibration_n_in_domain": 357,
+  "rff_dim": 1024,
+  "ridge": 0.01
+}

tokenizer.json ADDED Viewed

The diff for this file is too large to render. See raw diff

tokenizer_config.json ADDED Viewed

	@@ -0,0 +1,65 @@

+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": false,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "extra_special_tokens": {},
+  "mask_token": "[MASK]",
+  "max_length": 128,
+  "model_max_length": 512,
+  "never_split": null,
+  "pad_to_multiple_of": null,
+  "pad_token": "[PAD]",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "sep_token": "[SEP]",
+  "stride": 0,
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
+  "unk_token": "[UNK]"
+}

vocab.txt ADDED Viewed

The diff for this file is too large to render. See raw diff