d3rlpy.datasets.get_cartpole

d3rlpy.datasets.get_cartpole(as_tensor=False, device=None)

Returns cartpole dataset and environment.

The dataset is automatically downloaded to d3rlpy_data/cartpole.pkl if it does not exist.

Parameters:

• as_tensor (bool) – flag to hold observations as torch.Tensor.
• device (d3rlpy.gpu.device or int) – gpu deivce of device id for tensor.

Returns:

tuple of d3rlpy.dataset.MDPDataset and gym environment.

Return type:

tuple