hamzabouajila's picture
refactor the code for better scalability and update tsac naming to sentiment analysis, adding madar dataset for transliteration and normalization eval
bde1c71
raw
history blame contribute delete
716 Bytes
# 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,
}