Skip to content

Commit c1d7379

Browse files
author
Beat Buesser
committed
Update get_activations of TensorFlowV2Classifier
Signed-off-by: Beat Buesser <[email protected]>
1 parent b917a1b commit c1d7379

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

art/estimators/classification/tensorflow.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1301,9 +1301,10 @@ def get_activations(
13011301
i_layer = i_name
13021302
break
13031303
elif isinstance(layer, int):
1304-
if layer < 0 or layer >= len(self.layer_names):
1304+
if layer < -len(self.layer_names) or layer >= len(self.layer_names):
13051305
raise ValueError(
1306-
"Layer index %d is outside of range (0 to %d included)." % (layer, len(self.layer_names) - 1)
1306+
"Layer index %d is outside of range (-%d to %d)."
1307+
% (layer, len(self.layer_names), len(self.layer_names) - 1)
13071308
)
13081309
i_layer = layer
13091310
else:

0 commit comments

Comments
 (0)