Browse Source

Fix an api issue

tags/v1.2.1
jin-xiulang 4 years ago
parent
commit
5b24d1b7b2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindarmour/adv_robustness/evaluations/defense_evaluation.py

+ 1
- 1
mindarmour/adv_robustness/evaluations/defense_evaluation.py View File

@@ -106,7 +106,7 @@ class DefenseEvaluate:
Returns:
- float, the lower, the more successful the defense is.

- If return value == -1, len(idxes) == 0.
- If return value == -1, len(idxes) == 0.
"""
idxes = np.arange(self._num_samples)
cond1 = np.argmax(self._def_preds, axis=1) == self._true_labels


Loading…
Cancel
Save