Easy-to-use D4RL offline dataset loader
Project description
just-d4rl
Easy-to-use D4RL offline dataset loader
Installation
pip install just-d4rl
Usage
from just_d4rl import d4rl_offline_dataset
dataset = d4rl_offline_dataset("hopper-medium-v2")
dataset = d4rl_offline_dataset("walker2d-random-v2")
dataset = d4rl_offline_dataset("halfcheetah-medium-expert-v2")
dataset = d4rl_offline_dataset("antmaze-umaze-v2")
dataset = d4rl_offline_dataset("hopper-medium-v2")
dataset['observations'].shape, dataset['actions'].shape, dataset['rewards'].shape, dataset['terminals'].shape, dataset['timeouts'].shape
# ((1000000, 11), (1000000, 3), (1000000,), (1000000,), (1000000,))
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
just_d4rl-0.2407.1.tar.gz
(7.7 kB
view hashes)
Built Distribution
Close
Hashes for just_d4rl-0.2407.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3e3c7c63971eba69617f18c90523eb3c1cca37686fb90f888e593157cb43da3f |
|
MD5 | 174a722e580e77e4e72c921f04abc960 |
|
BLAKE2b-256 | 6ae19677ea3b3466550d3e404a738071b694b5d107ee4950b82bdee5c8227ed1 |