all taacks expet p attack

aidotse · Oct 16, 2024 · aa34e10 · aa34e10
1 parent 28bfc48
commit aa34e10
Show file tree

Hide file tree

Showing 6 changed files with 683 additions and 317 deletions.
diff --git a/examples/mia/cifar/audit.yml b/examples/mia/cifar/audit.yml
diff --git a/examples/mia/cifar/cifar_handler.py b/examples/mia/cifar/cifar_handler.py
@@ -8,11 +8,12 @@
 
 from leakpro import AbstractInputHandler
 
-class Cifar100InputHandler(AbstractInputHandler):
+class CifarInputHandler(AbstractInputHandler):
     """Class to handle the user input for the CIFAR100 dataset."""
 
     def __init__(self, configs: dict) -> None:
         super().__init__(configs = configs)
+        print(configs)
 
 
     def get_criterion(self)->None:
@@ -61,10 +62,6 @@ def train(
                 train_acc += pred.eq(labels.data.view_as(pred)).sum()
                 train_loss += loss.item()
 
-            log_train_str = (
-                f"Epoch: {epoch+1}/{epochs} | Train Loss: {train_loss/len(dataloader):.8f} | "
-                f"Train Acc: {float(train_acc)/len(dataloader.dataset):.8f}")
-            self.logger.info(log_train_str)
         model.to("cpu")
 
         return {"model": model, "metrics": {"accuracy": train_acc, "loss": train_loss}}
diff --git a/examples/mia/cifar/cifar_main.ipynb b/examples/mia/cifar/cifar_main.ipynb
diff --git a/examples/mia/cifar/utils/cifar_data_prepration.py b/examples/mia/cifar/utils/cifar_data_prepration.py
diff --git a/examples/mia/cifar/utils/cifar_model_prepration.py b/examples/mia/cifar/utils/cifar_model_prepration.py
diff --git a/examples/mia/tabular_mia/main.ipynb b/examples/mia/tabular_mia/main.ipynb