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