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

tf_agents.trajectories.policy_step.CommonFields

Strings which can be used for querying returned PolicyStep.info field.

For example, use getattr(info, CommonFields.LOG_PROBABILITY, None) to check if log probabilities are returned in the step or not.

LOG_PROBABILITY 'log_probability'