--- 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"