From 1f25ad8373191841646245e38e10c83815841572 Mon Sep 17 00:00:00 2001 From: marcsello Date: Sun, 15 Nov 2020 03:32:43 +0100 Subject: [PATCH] Attempt to reduce false positives --- src/preprocessor/soundpreprocessor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: