tf_agents.bandits.policies.constraints.InputNetworkConstraint

Class for representing a constraint using an input network.

Inherits From: BaseConstraint

This constraint class uses an input network to compute the action feasibility. It assumes that the input network is already trained and it can be provided at construction time or later using the set_network() function.

time_step_spec A TimeStep spec of the expected time_steps.
action_spec A nest of BoundedTensorSpec representing the actions.
input_network An instance of tf_agents.network.Network used to provide estimates of action feasibility.
name Python str name of this agent. All variables in this module will fall under that name. Defaults to the class name.

network

Methods

compute_loss

View source

__call__

View source

Returns the probability of input actions being feasible.