d3rlpy.datasets.get_atari¶
-
d3rlpy.datasets.
get_atari
(env_name)[source]¶ 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
Parameters: env_name (str) – environment id of d4rl-atari dataset. Returns: tuple of d3rlpy.dataset.MDPDataset
and gym environment.Return type: tuple