Skip to content

Commit

Permalink
Clarify documentation of labels_train/test usage wrt loss_train/test.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 665469121
  • Loading branch information
tensorflower-gardener committed Aug 20, 2024
1 parent 93c7e54 commit 89b3921
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -261,7 +261,9 @@ class AttackInputData:
# Contains ground-truth classes. For single-label classification, classes are
# assumed to be integers starting from 0. For multi-label classification,
# label is assumed to be multi-hot, i.e., labels is a binary array of shape
# (num_examples, num_classes).
# (num_examples, num_classes). Simplifies testing for classification models
# by computing the loss_train/test internally. Not used if loss_train/test is
# provided.
labels_train: Optional[np.ndarray] = None
labels_test: Optional[np.ndarray] = None

Expand Down

0 comments on commit 89b3921

Please sign in to comment.