Module: tf_agents.agents.categorical_dqn.categorical_dqn_agent

View source on GitHub

A Categorical DQN Agent.

Implements the Categorical DQN agent from

"A Distributional Perspective on Reinforcement Learning" Bellemare et al., 2017 https://arxiv.org/abs/1707.06887

Classes

class CategoricalDqnAgent: A Categorical DQN Agent based on the DQN Agent.

Functions

project_distribution(...): Projects a batch of (support, weights) onto target_support.