ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_privacy.make_keras_optimizer_class

Given a subclass of tf.keras.optimizers.Optimizer, returns a DP-SGD subclass of it.

cls Class from which to derive a DP subclass. Should be a subclass of tf.keras.optimizers.Optimizer.

A DP-SGD subclass of cls.