diff --git a/src/preprocessor/soundpreprocessor.py b/src/preprocessor/soundpreprocessor.py index a4324db..28744ad 100644 --- a/src/preprocessor/soundpreprocessor.py +++ b/src/preprocessor/soundpreprocessor.py @@ -130,7 +130,7 @@ class SoundPreProcessor(AbcPreProcessor): f"Sample {file_path} identified as {self._classes[class_id]} with the probablility of {probability[class_id]}" ) - return bool(class_id == self._target_id) + return bool((class_id == self._target_id) and (probability > 0.5)) def __del__(self): try: