diff --git a/mindarmour/fuzzing/fuzzing.py b/mindarmour/fuzzing/fuzzing.py index 2b8eb2395ee65210e9be24da00a545b71bae7060..af57bf278fb260a868d5e546fb0d9bff881194b7 100644 --- a/mindarmour/fuzzing/fuzzing.py +++ b/mindarmour/fuzzing/fuzzing.py @@ -393,13 +393,19 @@ class Fuzzer: temp = np.argmax(gt_labels, axis=1) == np.argmax(fuzz_preds, axis=1) metrics_report = {} if metrics == 'auto' or 'accuracy' in metrics: - acc = np.sum(temp) / np.size(temp) + if temp.any(): + acc = np.sum(temp) / np.size(temp) + else: + acc = 0 metrics_report['Accuracy'] = acc if metrics == 'auto' or 'attack_success_rate' in metrics: cond = [elem in self._attacks_list for elem in fuzz_strategies] temp = temp[cond] - attack_success_rate = 1 - np.sum(temp) / np.size(temp) + if temp.any(): + attack_success_rate = 1 - np.sum(temp) / np.size(temp) + else: + attack_success_rate = None metrics_report['Attack_success_rate'] = attack_success_rate if metrics == 'auto' or 'kmnc' in metrics or 'nbc' in metrics or 'snac' in metrics: