started unlearning setup

Tune.py

@@ -12,6 +12,8 @@ from datasets.UniversalIdentitySubset import UniversalIdentitySubset as Identity
 # models
 from architectures.Model import Model, Architecture
 
+from unlearning import LinearFiltration, WeightFiltration, CertifiedRemoval
+
 # numbre of classes 
 CLASS_SIZE = 20
 # batch 
@@ -145,3 +147,17 @@ print("Evaluating loaded")
 reloaded.evaluate(
     loader = test_loader
 )
+
+
+strategies_to_test = [
+    LinearFiltration(target_class_idx=12),
+    WeightFiltration(target_class_idx=12),
+    CertifiedRemoval(target_class_idx=12)
+]
+
+# Run the comparative benchmark seamlessly
+execution_profiles = {}
+for strategy in strategies_to_test:
+    # Each iteration clones weights back to fine-tuned state before running
+    runtime = my_model.unlearn(strategy, forget_loader, retain_loader)
+    execution_profiles[strategy.__class__.__name__] = runtime