Tinsae/Finetuning
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

attck metrics

Browse Source
This commit is contained in:
Tinsae
2026-07-08 00:25:07 +02:00
parent 026ca47800
commit eb8060fc05
37 changed files with 1649 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unlearning/WeightFiltration.py
View File

@@ -114,7 +114,7 @@ class WeightFiltration(Strategy):
model.eval()
if self.wf_model is None:
print(">> Initializing and compiling global WF-Net matrix (Run Once for all classes)...")
print("Initializing and compiling global WF-Net matrix (Run Once for all classes)...")
self.wf_model = self._optimise_filter(
model,
@@ -123,10 +123,10 @@ class WeightFiltration(Strategy):
device=device
)
else:
print(f">> Gating matrix loaded. Switching layout to target class index: {self.target_class_index}")
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
return self.wf_model.get()
def _split_data(self, dataset):
return vertical_split(