-
Notifications
You must be signed in to change notification settings - Fork 480
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
[Docs] new_dataset.md update #1827
Conversation
dict( | ||
abbr=f'lukaemon_mmlu_{_name}', | ||
type=MMLUDataset, | ||
path='opencompass/mmlu', |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
强调一下这里是个 mapping 的映射
- 接着,需要在`opencompass/utils/datasets_info.py`中创建对应名称的字典字段。如果用户已将数据集托管到huggingface或modelscope,那么请在`DATASETS_MAPPING`字典中添加对应名称的字段,并将对应的huggingface或modelscope数据集地址填入`ms_id`和`hf_id`;另外,还允许指定一个默认的`local`地址。具体示例如下: | ||
|
||
```python | ||
"opencompass/mmlu": { | ||
"ms_id": "opencompass/mmlu", | ||
"hf_id": "opencompass/mmlu", | ||
"local": "./data/mmlu/", | ||
} | ||
``` | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
用户需要在 load 函数中实现根据不同的环境变量切换数据源的操作,具体可以参考 cmmlu 中的实现
https://github.com/open-compass/opencompass/blob/main/opencompass/datasets/cmmlu.py。
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已添加
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
No description provided.