Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: nsl.keras

View source on GitHub

Keras APIs for Neural Structured Learning.

Modules

layers module: Keras layers for Neural Structured Learning.

Classes

class AdversarialRegularization: Wrapper thats adds adversarial regularization to a given tf.keras.Model.

class GraphRegularization: Class that wraps a given Keras model to include graph regularization.

Functions

adversarial_loss(...): Computes the adversarial loss for model given features and labels.