Add divide-by-zero guard

Oli-Ross · Feb 2, 2024 · 89354d4 · 89354d4
1 parent 6be41e5
commit 89354d4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mia/attack_pipeline.py b/mia/attack_pipeline.py
@@ -105,7 +105,8 @@ def run_pipeline(targetModel, targetTrainData, targetRestData):
         if nonmemberAttackPrediction:
             membersInferredAsMembers = len(memberAttackPrediction) - np.count_nonzero(memberAttackPrediction)
             nonmembersInferredAsMembers = len(nonmemberAttackPrediction) - np.count_nonzero(nonmemberAttackPrediction)
-            precisionPerClass[_class] = membersInferredAsMembers / (membersInferredAsMembers + nonmembersInferredAsMembers)
+            if (membersInferredAsMembers + nonmembersInferredAsMembers):
+                precisionPerClass[_class] = membersInferredAsMembers / (membersInferredAsMembers + nonmembersInferredAsMembers)
 
     membersInferredAsMembers = targetTrainDataSize - sum([sum(x) for x in memberAttackPredictions])
     nonmembersInferredAsMembers = targetTrainDataSize - sum([sum(x) for x in nonmemberAttackPredictions])