626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Update per-class descriptions for DP Keras Model classes.

Browse source 
PiperOrigin-RevId: 367515250
This commit is contained in:
Steve Chien 2021-04-08 15:06:12 -07:00 committed by A. Unique TensorFlower
parent 121982deb1
commit 3c64cce796
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tensorflow_privacy/privacy/keras_models/dp_keras_model.py
View file

@ -36,7 +36,7 @@ def make_dp_model_class(cls):
gradients). gradients).
noise_multiplier: Ratio of the standard deviation to the clipping noise_multiplier: Ratio of the standard deviation to the clipping
norm. norm.
use_xla: If True, compiles train_step to XLA. use_xla: If `True`, compiles train_step to XLA.
""" """
super(DPModelClass, self).__init__(*args, **kwargs) super(DPModelClass, self).__init__(*args, **kwargs)
self._l2_norm_clip = l2_norm_clip self._l2_norm_clip = l2_norm_clip
@ -87,6 +87,8 @@ def make_dp_model_class(cls):
self.compiled_metrics.update_state(y, y_pred) self.compiled_metrics.update_state(y, y_pred)
return {m.name: m.result() for m in self.metrics} return {m.name: m.result() for m in self.metrics}
DPModelClass.__doc__ = ('DP subclass of `tf.keras.{}`.').format(cls.__name__)
return DPModelClass return DPModelClass