attcks

2026-07-08 10:07:39 +02:00
parent eb8060fc05
commit a118a975c1
20 changed files with 287 additions and 22 deletions
--- a/unlearning/WeightFiltration.py
+++ b/unlearning/WeightFiltration.py
@@ -126,7 +126,7 @@ class WeightFiltration(Strategy):
            print(f"Gating matrix loaded. Switching layout to target class index: {self.target_class_index}")
            self.wf_model.target_class_index = self.target_class_index
            
-        return self.wf_model.get()
+        return self.wf_model
    
    def _split_data(self, dataset):
        return vertical_split(