Attempt to reduce false positives
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Pünkösd Marcell 2020-11-15 03:32:43 +01:00
parent 11e0aaf0c3
commit 1f25ad8373

View File

@ -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: