Former-commit-id: 3404e8f302d0b0308825fccedec82d8781b8c39e
This commit is contained in:
hiyouga 2024-05-30 00:52:26 +08:00
parent 9b6bdf9449
commit 107e39f2de

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")):