Module: tf_agents.trajectories.policy_step

View source on GitHub

Policy Step.

Classes

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

class PolicyInfo: PolicyInfo(log_probability,)

class PolicyStep: PolicyStep(action, state, info)

Functions

get_log_probability(...): Gets the CommonFields.LOG_PROBABILITY from info depending on type.

set_log_probability(...): Sets the CommonFields.LOG_PROBABILITY on info to be log_probability.