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 details)
Built Distribution
File details
Details for the file just_d4rl-0.2407.1.tar.gz
.
File metadata
- Download URL: just_d4rl-0.2407.1.tar.gz
- Upload date:
- Size: 7.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.12.2
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4f175d914c64df453f5be1ceb80924ee47626b60c62078c2295e149829c3d075 |
|
MD5 | 9bd2e38cf6cbeb4a90b59ebe125c0d4f |
|
BLAKE2b-256 | d436b042f15a23f1587f537dd296e89dec66b71f8fd0e51c85102193d48cb40a |
File details
Details for the file just_d4rl-0.2407.1-py3-none-any.whl
.
File metadata
- Download URL: just_d4rl-0.2407.1-py3-none-any.whl
- Upload date:
- Size: 6.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.12.2
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3e3c7c63971eba69617f18c90523eb3c1cca37686fb90f888e593157cb43da3f |
|
MD5 | 174a722e580e77e4e72c921f04abc960 |
|
BLAKE2b-256 | 6ae19677ea3b3466550d3e404a738071b694b5d107ee4950b82bdee5c8227ed1 |