selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor update

Browse Source
This commit is contained in:
Gokul Mohanarangan
2023-08-10 14:37:39 +05:30
parent af954e2818
commit 992134a38b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
server/evaluate/evaluate_transcription.py
View File

@@ -49,7 +49,7 @@ class TestDatasetLoader(BaseModel):
def _load_test_data(self) -> tuple[Path, Path]: def _load_test_data(self) -> tuple[Path, Path]:
""" """
Loader function to validate inout files and generate samples Loader function to validate input files and generate samples
""" """
PREDICTED_TEST_SAMPLES_DIR = self.test_dir / "predicted_texts" PREDICTED_TEST_SAMPLES_DIR = self.test_dir / "predicted_texts"
REFERENCE_TEST_SAMPLES_DIR = self.test_dir / "reference_texts" REFERENCE_TEST_SAMPLES_DIR = self.test_dir / "reference_texts"
@@ -116,6 +116,7 @@ class ModelEvaluator:
Returns the parameters defining the evaluator Returns the parameters defining the evaluator
""" """
return self.evaluation_config.model_dump() return self.evaluation_config.model_dump()
def _normalize(self, sample: EvaluationTestSample) -> None: def _normalize(self, sample: EvaluationTestSample) -> None:
""" """
Normalize both reference and predicted text Normalize both reference and predicted text