binary test output in working state

util/module_mixins.py

@@ -163,7 +163,8 @@ class TestMixin:
                            enumerate(['background', 'chimpanze', 'geunon', 'mandrille', 'redcap'])}
         else:
             pred = torch.stack([x.mean(dim=0) if x.shape[0] > 1 else x for x in sorted_y.values()]).squeeze()
-            class_names = {val: key for val, key in ['negative', 'positive']}
+            pred = torch.where(pred > 0.5, 1, 0)
+            class_names = {val: key for val, key in enumerate(['negative', 'positive'])}
 
 
         df = pd.DataFrame(data=dict(filename=[Path(x).name for x in sorted_y.keys()],