Skip to content

Commit 068fba5

Browse files
Irina NicolaeIrina Nicolae
authored andcommitted
Update doc in classifiers
(cherry picked from commit 9e341f7)
1 parent 39ccfe1 commit 068fba5

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

art/classifiers/classifier.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
import abc
44
import sys
55

6+
# TODO Add tests for defences on classifier
7+
68
# Ensure compatibility with Python 2 and 3 when using ABCMeta
79
if sys.version_info >= (3, 4):
810
ABC = abc.ABC
@@ -20,6 +22,8 @@ def __init__(self, clip_values, defences=None):
2022
:param clip_values: Tuple of the form `(min, max)` representing the minimum and maximum values allowed
2123
for features.
2224
:type clip_values: `tuple`
25+
:param defences: Defences to be activated with the classifier.
26+
:type defences: `str` or `list(str)`
2327
"""
2428
self._clip_values = clip_values
2529
self._parse_defences(defences)

art/classifiers/keras.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,10 @@ def __init__(self, clip_values, model, use_logits=False, defences=None):
1818
:type clip_values: `tuple`
1919
:param model: Keras model
2020
:type model: `keras.models.Sequential`
21-
:param use_logits: True if the output of the model are the logits
21+
:param use_logits: True if the output of the model are the logits.
2222
:type use_logits: `bool`
23+
:param defences: Defences to be activated with the classifier.
24+
:type defences: `str` or `list(str)`
2325
"""
2426
import keras.backend as k
2527

0 commit comments

Comments
 (0)