d3rlpy.datasets.get_d4rl

d3rlpy.datasets.get_d4rl(env_name, create_mask=False, mask_size=1)

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

• Fu et al., D4RL: Datasets for Deep Data-Driven Reinforcement Learning.

• https://github.com/rail-berkeley/d4rl

Parameters

• env_name (str) – environment id of d4rl dataset.

• create_mask (bool) – flag to create binary mask for bootstrapping.

• mask_size (int) – ensemble size for binary mask.

Returns

tuple of d3rlpy.dataset.MDPDataset and gym environment.

Return type

Tuple[d3rlpy.dataset.MDPDataset, gym.core.Env]