Method load

opencompass/datasets/mgsm.py:16–29

(path: str)

    @staticmethod
    def load(path: str):
        path = get_data_path(path, local_mode=True)
        src_lines = open(path, 'r', encoding='utf-8').readlines()
        data = {'question': [], 'answer': []}
        for lines in src_lines:
            question, answer = lines.strip().split('\t')
            data['question'].append(question)
            data['answer'].append(answer)

        dataset = Dataset.from_dict({
            'question': data['question'],
            'answer': data['answer']
        })
        return dataset


LANG_TO_ANSWER_PREFIX = {