Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] AGIEval数据集是否可以支持few-shot评测设置? #1150

Open
1 task
cabbagecabbage opened this issue May 14, 2024 · 0 comments
Open
1 task
Assignees

Comments

@cabbagecabbage
Copy link

描述该功能

opencompass/opencompass/datasets/agieval/agieval.py 文件显示,目前AGIEval评测仅支持zero-shot设置:

def load(path: str, name: str, setting_name: str):
      from .dataset_loader import load_dataset, load_dataset_as_result_schema

      assert setting_name in 'zero-shot', 'only support zero-shot setting'
      dataset_wo_label = load_dataset(name, setting_name, path)
      dataset_with_label = load_dataset_as_result_schema(name, path)
      dataset = []
      for d1, d2 in zip(dataset_wo_label, dataset_with_label):
          dataset.append({
              'id': d2.index,
              'problem_input': d1['context'],
              'label': d2.label,
          })
      dataset = Dataset.from_list(dataset)
      return dataset

原生AGIEval支持few-shot设置,是否可以追加few-shot评测设置?

是否希望自己实现该功能?

  • 我希望自己来实现这一功能,并向 OpenCompass 贡献代码!
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants