ColossalAI/applications/ColossalEval/colossal_eval/dataset/__init__.py

28 lines
741 B
Python
Raw Normal View History

from .agieval import AGIEvalDataset
from .base import BaseDataset
from .ceval import CEvalDataset
from .cmmlu import CMMLUDataset
from .colossalai import ColossalDataset
from .gaokaobench import GaoKaoBenchDataset
from .longbench import LongBenchDataset
from .mmlu import MMLUDataset
from .mtbench import MTBenchDataset
from .safetybench_en import SafetyBenchENDataset
from .safetybench_zh import SafetyBenchZHDataset
from .cvalues import CValuesDataset
__all__ = [
"AGIEvalDataset",
"BaseDataset",
"CEvalDataset",
"CMMLUDataset",
"GaoKaoBenchDataset",
"LongBenchDataset",
"MMLUDataset",
"ColossalDataset",
"MTBenchDataset",
'SafetyBenchENDataset',
'SafetyBenchZHDataset',
'CValuesDataset'
]