File size: 365 Bytes
bde1c71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# evaluators/base_evaluator.py
from abc import ABC, abstractmethod
from typing import Dict, Any

class BaseEvaluator(ABC):
    @abstractmethod
    def load_dataset(self):
        pass

    @abstractmethod
    def evaluate(self, model, tokenizer, device) -> Dict[str, Any]:
        pass

    @property
    @abstractmethod
    def task_name(self) -> str:
        pass