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

Module: tf_agents.bandits.policies.constraints

An API for representing constraints.

Classes

class AbsoluteConstraint: Class for representing a trainable absolute value constraint.

class BaseConstraint: Abstract base class for representing constraints.

class InputNetworkConstraint: Class for representing a constraint using an input network.

class NeuralConstraint: Class for representing a trainable constraint using a neural network.

class QuantileConstraint: Class for representing a trainable quantile constraint.

class RelativeConstraint: Class for representing a trainable relative constraint.

class RelativeQuantileConstraint: Class for representing a trainable relative quantile constraint.

Functions

compute_feasibility_probability(...): Helper function to compute the action feasibility probability.

construct_mask_from_multiple_sources(...): Constructs an action mask from multiple sources.