d3rlpy.datasets.get_pendulum¶

d3rlpy.datasets.get_pendulum(dataset_type='replay', transition_picker=None, trajectory_slicer=None, render_mode=None)[source]¶

Returns pendulum dataset and environment.

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

Parameters:

dataset_type (str) – dataset type. Available options are ['replay', 'random'].
transition_picker (Optional[TransitionPickerProtocol]) – TransitionPickerProtocol object.
trajectory_slicer (Optional[TrajectorySlicerProtocol]) – TrajectorySlicerProtocol object.
render_mode (Optional[str]) – Mode of rendering (human, rgb_array).

Returns:

tuple of d3rlpy.dataset.ReplayBuffer and gym environment.

Return type: