Skip to content

Commit 01babca

Browse files
csferngtensorflow-copybara
authored andcommitted
Update docstring for AdversarialRegularization.perturb_on_batch.
PiperOrigin-RevId: 315319698
1 parent c227ea0 commit 01babca

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

neural_structured_learning/keras/adversarial_regularization.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -700,7 +700,8 @@ def perturb_on_batch(self, x, **config_kwargs):
700700
**config_kwargs: (optional) hyperparameters for generating adversarial
701701
preturbation. Any keyword argument here will overwrite the corresponding
702702
field in `nsl.configs.AdvNeighborConfig` specified in `__init__`.
703-
Acceptable keys: `feature_mask`, `adv_step_size`, and `adv_grad_norm`.
703+
Acceptable keys: `feature_mask`, `adv_step_size`, `adv_grad_norm`,
704+
`clip_value_min`, `clip_value_max`, `pgd_iterations`, and `pgd_epsilon`.
704705
705706
Returns:
706707
A dictionary of NumPy arrays, `SparseTensor`, or `RaggedTensor` objects of

0 commit comments

Comments
 (0)