DatasetsΒΆ

d3rlpy provides datasets for experimenting data-driven deep reinforcement learning algorithms.

d3rlpy.datasets.get_cartpole

Returns cartpole dataset and environment.

d3rlpy.datasets.get_pendulum

Returns pendulum dataset and environment.

d3rlpy.datasets.get_atari

Returns atari dataset and envrironment.

d3rlpy.datasets.get_atari_transitions

Returns atari dataset as a list of Transition objects and envrironment.

d3rlpy.datasets.get_d4rl

Returns d4rl dataset and envrironment.

d3rlpy.datasets.get_dataset

Returns dataset and envrironment by guessing from name.

d3rlpy.datasets.get_minari

Returns minari dataset and envrironment.