d3rlpy.datasets.get_cartpole

d3rlpy.datasets.get_cartpole(dataset_type='replay', transition_picker=None, trajectory_slicer=None)

Returns cartpole dataset and environment.

The dataset is automatically downloaded to d3rlpy_data/cartpole.h5 if it does not exist.

Parameters

• dataset_type (str) – dataset type. Available options are ['replay', 'random'].

• transition_picker (Optional[d3rlpy.dataset.transition_pickers.TransitionPickerProtocol]) – TransitionPickerProtocol object.

• trajectory_slicer (Optional[d3rlpy.dataset.trajectory_slicers.TrajectorySlicerProtocol]) – TrajectorySlicerProtocol object.

Returns

tuple of d3rlpy.dataset.ReplayBuffer and gym environment.

Return type

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