Create config.json

config.json

@@ -0,0 +1,89 @@
+{
+    "base_model": "microsoft/DeBERTa-v3-base",
+    "config_path": null,
+    "fc_dropout": 0.2,
+    "targets": [
+        "task_type_1",
+        "task_type_2",
+        "task_type_prob",
+        "creativity_scope",
+        "reasoning",
+        "contextual_knowledge",
+        "number_of_few_shots",
+        "domain_knowledge",
+        "no_label_reason",
+        "constraint_ct",
+        "prompt_complexity_score"
+    ],
+    "target_sizes": {
+      "task_type": 12,
+      "creativity_scope": 3,
+      "reasoning": 2,
+      "contextual_knowledge": 2,
+      "number_of_few_shots": 6,
+      "domain_knowledge": 4,
+      "no_label_reason": 1,
+      "constraint_ct": 2
+    },
+    "task_type_map": {
+        "0": "Brainstorming",
+        "1": "Chatbot",
+        "2": "Classification",
+        "3": "Closed QA",
+        "4": "Code Generation",
+        "5": "Extraction",
+        "6": "Open QA",
+        "7": "Other",
+        "8": "Rewrite",
+        "9": "Summarization",
+        "10": "Text Generation",
+        "11": "Unknown"
+    },
+    "weights_map": {
+        "creativity_scope": [2, 1, 0],
+        "reasoning": [0, 1],
+        "contextual_knowledge": [0, 1],
+        "number_of_few_shots": [0, 1, 2, 3, 4, 5],
+        "domain_knowledge": [3, 1, 2, 0],
+        "no_label_reason": [0],
+        "constraint_ct": [1, 0]
+    },
+    "divisor_map": {
+        "creativity_scope": 2,
+        "reasoning": 1,
+        "contextual_knowledge": 1,
+        "number_of_few_shots": 1,
+        "domain_knowledge": 3,
+        "no_label_reason": 1,
+        "constraint_ct": 1
+    },
+    "model_output_type": {
+        "task_type_1": "string",
+        "task_type_2": "string",
+        "task_type_prob": "numeric",
+        "creativity_scope": "numeric",
+        "reasoning": "numeric",
+        "contextual_knowledge": "numeric",
+        "number_of_few_shots": "numeric",
+        "domain_knowledge": "numeric",
+        "no_label_reason": "numeric",
+        "constraint_ct": "numeric",
+        "prompt_complexity_score": "numeric"
+    },
+    "creativity_score_map": {"High": 0, "Low": 1, "No": 2},
+    "reasoning_map": {"Unknown": -1, "No": 0, "Yes": 1},
+    "contextual_knowledge_map": {"Unknown": -1, "No": 0, "Yes": 1},
+    "number_of_few_shots_map": {
+      "0.0": 0,
+      "1.0": 1,
+      "2.0": 2,
+      "3.0": 3,
+      "4.0": 4,
+      "5.0": 5
+    },
+    "domain_knowledge_map": {"High": 0, "Low": 1, "Medium": 2, "No": 3},
+    "no_label_reason_map": {"Unknown": 0},
+    "constraint_ct_map": {"1.0": 0, "Unknown": 1},
+    "pretrained": true,
+    "drop_out": false
+}