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

30 lines
788 B
Python

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