utils_medmcqa.py 541 Bytes
Newer Older
Tanishq Abraham's avatar
Tanishq Abraham committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# Copied from Master
def doc_to_text(doc) -> str:
    """
    Question: <question>
    Choices:
    A. <choice1>
    B. <choice2>
    C. <choice3>
    D. <choice4>
    Answer:
    """
    choices = [doc["opa"], doc["opb"], doc["opc"], doc["opd"]]
    option_choices = {'A': choices[0], 'B': choices[1], 'C': choices[2], 'D': choices[3]}

    prompt = "Question: " + doc["question"] + "\nChoices:\n"
    for choice, option in option_choices.items():
        prompt += f"{choice.upper()}. {option}\n"
    prompt += "Answer:"
    return prompt