d3rlpy.datasets.get_atari¶
-
d3rlpy.datasets.
get_atari
(env_name, as_tensor=False, device=None)[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: Returns: tuple of
d3rlpy.dataset.MDPDataset
and gym environment.Return type: