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