d3rlpy.datasets.get_atari

d3rlpy.datasets.get_atari(env_name, num_stack=None, render_mode=None)

Returns atari dataset and envrironment.

The dataset is provided through d4rl-atari. See more details including available dataset from its GitHub page.

from d3rlpy.datasets import get_atari

dataset, env = get_atari('breakout-mixed-v0')

References

• https://github.com/takuseno/d4rl-atari

Parameters

• env_name (str) – environment id of d4rl-atari dataset.

• num_stack (Optional[int]) – the number of frames to stack (only applied to env).

• render_mode (Optional[str]) – Mode of rendering (human, rgb_array).

Returns

tuple of d3rlpy.dataset.ReplayBuffer and gym environment.

Return type

Tuple[d3rlpy.dataset.replay_buffer.ReplayBuffer, gym.core.Env[numpy.ndarray, int]]