File size: 716 Bytes
bde1c71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# src/evaluators/__init__.py
from typing import Dict, Type
from .base_evaluator import BaseEvaluator

# Import all evaluators
from .sentiment_analysis.evaluator import SentimentAnalysisEvaluator
# from .tunisian_corpus_coverage import TunisianCorpusCoverageEvaluator
# Add new ones here as you create them:
from .normalization import NormalizationEvaluator
from .transliteration import TransliterationEvaluator

# Registry: task_name → Evaluator class
EVALUATOR_REGISTRY: Dict[str, Type[BaseEvaluator]] = {
    "Sentiment Analysis": SentimentAnalysisEvaluator,
    # "Corpus Coverage": TunisianCorpusCoverageEvaluator,
    "Normalization": NormalizationEvaluator,
    "Transliteration": TransliterationEvaluator,
}