From 1d7d1274190a95e63e7c8bb4c3727ad5cc721b9f Mon Sep 17 00:00:00 2001 From: marcsello Date: Sun, 15 Nov 2020 04:20:39 +0100 Subject: [PATCH] Fixed the previous fix --- 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 28744ad..b43b3fa 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) and (probability > 0.5)) + return bool((class_id == self._target_id) and (probability[class_id] > 0.5)) def __del__(self): try: