preprocess_pubmedqa.py

def doc_to_text(doc):
    ctxs = "\n".join(doc["context"]["contexts"])
    return "Abstract: {}\nQuestion: {}\nAnswer:".format(
        ctxs, doc["question"], doc["final_decision"]
    )

def doc_to_target(doc):
    return " {}".format(doc["final_decision"])

def gold_alias(doc):
    dict_to_label = {
        'yes': 0,
        'no': 1,
        'maybe': 2
    }
    return dict_to_label[doc["final_decision"]]