| def doc_to_target(doc): | |
| labels = [c["label"] for c in doc["question"]["choices"]] | |
| try: | |
| i = labels.index(doc["answerKey"].lstrip()) | |
| except Exception as e: | |
| print("Failed", e) | |
| return | |
| return i | |
| def doc_to_choice(doc): | |
| texts = [c["text"] for c in doc["question"]["choices"]] | |
| return texts | |
| def doc_to_text(doc): | |
| return doc["question"]["stem"].strip() | |