d3rlpy.datasets.get_pendulum

d3rlpy.datasets.get_pendulum(create_mask=False, mask_size=1, dataset_type='replay')[source]

Returns pendulum dataset and environment.

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

Parameters

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

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

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

Returns

tuple of d3rlpy.dataset.MDPDataset and gym environment.

Return type

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