add smooething

TheCacophonyProject · Oct 1, 2024 · c3ce590 · c3ce590
1 parent 1e67f11
commit c3ce590
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/modelevaluate.py b/src/modelevaluate.py
@@ -462,7 +462,7 @@ def evaluate_dir(
                 #     top_score = len(output)
                 #     smoothed = output
                 # else:
-                # smoothed = output * output * masses
+                smoothed = output * output * masses
                 prediction.classified_clip(output, output, data[2], top_score=top_score)
                 y_true.append(label_mapping.get(label, label))
                 predicted_labels = [prediction.predicted_tag()]