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

more lint

Browse source
This commit is contained in:
npapernot 2019-07-29 22:22:52 +00:00
parent b1f1c47cae
commit 801e6eef1c
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
privacy/bolton/models.py
View file

@ -86,7 +86,7 @@ class BoltonModel(Model): # pylint: disable=abstract-method
loss: The loss function to use. Must be a StrongConvex loss (extend the loss: The loss function to use. Must be a StrongConvex loss (extend the
StrongConvexMixin). StrongConvexMixin).
kernel_initializer: The kernel initializer to use for the single layer. kernel_initializer: The kernel initializer to use for the single layer.
kwargs: kwargs to keras Model.compile. See super. **kwargs: kwargs to keras Model.compile. See super.
""" """
if not isinstance(loss, StrongConvexMixin): if not isinstance(loss, StrongConvexMixin):
raise ValueError('loss function must be a Strongly Convex and therefore ' raise ValueError('loss function must be a Strongly Convex and therefore '
@ -136,7 +136,7 @@ class BoltonModel(Model): # pylint: disable=abstract-method
See the bolton paper for more description. See the bolton paper for more description.
noise_distribution: the distribution to pull noise from. noise_distribution: the distribution to pull noise from.
steps_per_epoch: steps_per_epoch:
kwargs: kwargs to keras Model.fit. See super. **kwargs: kwargs to keras Model.fit. See super.
Returns: Returns:
output output
@ -201,6 +201,9 @@ class BoltonModel(Model): # pylint: disable=abstract-method
n_samples: number of individual samples in x n_samples: number of individual samples in x
steps_per_epoch: steps_per_epoch:
**kwargs: **kwargs **kwargs: **kwargs
Returns:
output
""" """
if class_weight is None: if class_weight is None:
class_weight = self.calculate_class_weights(class_weight) class_weight = self.calculate_class_weights(class_weight)