xpag.agents.flax_agents.sdqn.sdqn.FlaxSDQN#

class FlaxSDQN(observation_dim, action_dim, params=None)#

Bases: Agent

Jax implementation of SDQN (https://arxiv.org/pdf/1705.05035.pdf).

Methods

`load`
`save`
`select_action`
`train_on_batch`
`write_config`