cleaned up for submission

2026-07-10 20:13:13 +02:00
parent 47fbb19761
commit 7eff030e89
38 changed files with 39 additions and 4175 deletions
--- a/eval/UnlearningAttack.py
+++ b/eval/UnlearningAttack.py
@@ -32,7 +32,8 @@ class UnlearningAttack:
                if model1.__class__.__name__ == "WF_Module":
                    gate = torch.full((data.size(0),), target_class, device=device)
                    p1 = torch.softmax(model1(data, target_class_indices=gate), dim=1)
-                else:
+                    
+                else: # its a normal nn.Module
                    p1 = torch.softmax(model1(data), dim=1)
                p2 = torch.softmax(model2(data), dim=1)
                probs1.extend(p1.cpu().numpy()); probs2.extend(p2.cpu().numpy())