Fix typo

Oli-Ross · Feb 24, 2024 · c4c37a5 · c4c37a5
1 parent 7e5e8be
commit c4c37a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mia/shadow_models.py b/mia/shadow_models.py
@@ -162,11 +162,11 @@ def evaluate_models(models:List[tm.Sequential], datasets: List[Tuple[ds.Dataset,
     hash = utils.hash(str(config))
 
     with open(f"{hash}_shadowModelTrainAccuracy.csv",'w') as file:
-        file.write(f"Attack Model Training Accuracies (Overall:{np.average(train_accuracies)})\n")
+        file.write(f"Shadow Model Training Accuracies (Overall:{np.average(train_accuracies)})\n")
         for train_acc in train_accuracies:
             file.write(f"{train_acc}\n")
     with open(f"{hash}_shadowModelTestAccuracy.csv",'w') as file:
-        file.write(f"Attack Model Testing Accuracies (Overall:{np.average(test_accuracies)})\n")
+        file.write(f"Shadow Model Testing Accuracies (Overall:{np.average(test_accuracies)})\n")
         for test_acc in test_accuracies:
             file.write(f"{test_acc}\n")