Former-commit-id: 72965aa3f13a9c085c29781b6790d80d00a545d8
This commit is contained in:
hiyouga 2024-05-30 00:52:26 +08:00
parent 4f7c850115
commit 14ab14a0e6

View File

@ -154,7 +154,7 @@ class MMLU(datasets.GeneratorBasedBuilder):
]
def _generate_examples(self, filepath):
df = pd.read_csv(filepath)
df = pd.read_csv(filepath, header=None)
df.columns = ["question", "A", "B", "C", "D", "answer"]
for i, instance in enumerate(df.to_dict(orient="records")):