d3rlpy.datasets.get_d4rl¶
- d3rlpy.datasets.get_d4rl(env_name, transition_picker=None, trajectory_slicer=None, render_mode=None)[source]¶
Returns d4rl dataset and envrironment.
The dataset is provided through d4rl.
from d3rlpy.datasets import get_d4rl dataset, env = get_d4rl('hopper-medium-v0')
References
- Parameters:
env_name (str) – environment id of d4rl dataset.
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:
Tuple[ReplayBuffer, Env[ndarray[Any, dtype[Any]], ndarray[Any, dtype[Any]]]]