__init__.py 281 B

1234567
  1. from .configs import *
  2. from .model import ModelForEvaluation
  3. from .tasks import BaseTask, GenerationTask, MultiChoiceTask
  4. from .metrics import qa_evaluate
  5. from .utils import print_rank_0
  6. DEFAULT_CLASS = {TaskType.GENERATION: GenerationTask, TaskType.MULTICHOICE: MultiChoiceTask}