Diff of /src/constants/ddi.py [000000] .. [735bb5]

Switch to side-by-side view

--- a
+++ b/src/constants/ddi.py
@@ -0,0 +1,45 @@
+# coding: utf-8
+
+# Base Dependencies
+# ------------------
+from pathlib import Path
+from os.path import join as pjoin
+
+
+DDI_VOCAB_PATH = Path(pjoin("data", "ddi", "vocab", "vocab.txt"))
+DDI_PATH = Path("data/ddi")
+DDI_ENTITY_TYPES = ["DRUG", "GROUP", "BRAND", "DRUG_N"]
+DDI_NO_REL = "NO-REL"
+DDI_REL_TYPES = ["EFFECT", "MECHANISM", "ADVISE", "INT"]
+DDI_ALL_TYPES = [DDI_NO_REL] + DDI_REL_TYPES
+DDI_REL_TEST_COUNTS ={"EFFECT": 360, "MECHANISM": 302, "ADVISE": 221, "INT": 96}
+DDI_REL_TEST_WEIGHTS = [
+    360/979,
+    302/979,
+    221/979,
+    96/979,
+]
+
+DDI_ATTR_ENTITY_CANDIDATES = {
+    "DRUG": DDI_ENTITY_TYPES,
+    "GROUP": DDI_ENTITY_TYPES,
+    "BRAND": DDI_ENTITY_TYPES,
+    "DRUG_N": DDI_ENTITY_TYPES,
+}
+
+DDI_IOB_TAGS = [
+    "O",
+    "B-DRUG",
+    "I-DRUG",
+    "B-GROUP",
+    "I-GROUP",
+    "B-BRAND",
+    "I-BRAND",
+    "B-DRUG_N",
+    "I-DRUG_N",
+]
+
+DDI_RD_MAX = 20
+
+DDI_HF_TRAIN_PATH = "data/ddi/train.hf"
+DDI_HF_TEST_PATH = "data/ddi/test.hf"