Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
utils.py 361 Bytes
Newer Older
lintangsutawika's avatar
fixed super_glue and removed unused yaml config  
lintangsutawika committed
1 
def doc_to_text(doc):
FarzanehNakhaee's avatar
black update  
FarzanehNakhaee committed
2
3
4
5 
    return "Sentence 1: {}\nSentence 2: {}\nQuestion: Is the word '{}' used in the same way in the" " two sentences above?\nAnswer:".format(
        doc["sentence1"],
        doc["sentence2"],
        doc["sentence1"][doc["start1"] : doc["end1"]],
lintangsutawika's avatar
fixed super_glue and removed unused yaml config  
lintangsutawika committed
6
7
8
9
10 
    )


def doc_to_target(doc):
    return " {}".format({0: "no", 1: "yes"}[doc["label"]])