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

Module: nsl.estimator

Estimator APIs for Neural Structured Learning.

The current NSL Estimator APIs may not be compatible with certain TensorFlow versions (such as 2.0 or above).

Functions

add_adversarial_regularization(...): Adds adversarial regularization to a tf.estimator.Estimator.

add_graph_regularization(...): Adds graph regularization to a tf.estimator.Estimator.