wf_net

Tune_new.py

@@ -13,6 +13,8 @@ from unlearning.CertifiedRemoval import CertifiedRemoval
 from unlearning.CertifiedUnlearning import CertifiedUnlearning
 from unlearning.LinearFiltration import LinearFiltration
 from unlearning.WeightFiltration import WeightFiltration
+from unlearning.WF import WeightF
+
 
 # Global Hyperparameters
 CLASS_SIZE = 20
@@ -255,16 +257,16 @@ if __name__ == "__main__":
             target_class_index=i
         )
 
-        weight_filtration = WeightFiltration(
+        weight_filtration = WeightF( #WeightFiltration(
             target_class_index=i,
             epochs=3,
-            lr=0.5,
-            gamma=150
+            lr=0.05,
+            gamma=5
         )
 
         strategies = [
-            certified_unlearning,
-            # weight_filtration,
+            # certified_unlearning,
+            weight_filtration,
             # linear_filtration
         ]