d3rlpy.datasets.get_atari_transitions¶
- d3rlpy.datasets.get_atari_transitions(game_name, fraction=0.01, index=0, num_stack=None, render_mode=None)[source]¶
Returns atari dataset as a list of Transition objects and envrironment.
The dataset is provided through d4rl-atari. The difference from
get_atari
function is that this function will sample transitions from all epochs. This function is necessary for reproducing Atari experiments.from d3rlpy.datasets import get_atari_transitions # get 1% of transitions from all epochs (1M x 50 epoch x 1% = 0.5M) dataset, env = get_atari_transitions('breakout', fraction=0.01)
References
- Parameters
game_name (str) – Atari 2600 game name in lower_snake_case.
fraction (float) – fraction of sampled transitions.
index (int) – index to specify which trial to load.
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 a list of
d3rlpy.dataset.Transition
and gym environment.- Return type
Tuple[d3rlpy.dataset.replay_buffer.ReplayBuffer, gym.core.Env[numpy.ndarray, int]]