This commit is contained in:
hiyouga
2024-05-30 00:52:26 +08:00
parent 483eb47e5d
commit 3404e8f302

View File

@@ -154,7 +154,7 @@ class MMLU(datasets.GeneratorBasedBuilder):
] ]
def _generate_examples(self, filepath): 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"] df.columns = ["question", "A", "B", "C", "D", "answer"]
for i, instance in enumerate(df.to_dict(orient="records")): for i, instance in enumerate(df.to_dict(orient="records")):