d3rlpy.datasets.get_cartpole¶
- d3rlpy.datasets.get_cartpole(dataset_type='replay', transition_picker=None, trajectory_slicer=None, render_mode=None)[source]¶
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.
render_mode (Optional[str]) – Mode of rendering (
human
,rgb_array
).
- Returns
tuple of
d3rlpy.dataset.ReplayBuffer
and gym environment.- Return type
Tuple[d3rlpy.dataset.replay_buffer.ReplayBuffer, gym.core.Env[numpy.ndarray[Any, numpy.dtype[Any]], int]]