Easy-to-use D4RL offline dataset loader
Project description
just-d4rl
Easy-to-use D4RL offline dataset loader, focused solely on downloading and providing D4RL datasets without dependencies on gym or gymnasium.
Key Features
- Downloads and provides D4RL offline datasets
- No dependencies on gym or gymnasium
- Lightweight and focused functionality
- Easy integration with PyTorch
Installation
Install from PyPI:
pip install just-d4rl
Usage
from just_d4rl import D4RLDataset, d4rl_offline_dataset, D4RLScoreNormalizer
# Download and load a D4RL dataset
d4rl_dataset = d4rl_offline_dataset("hopper-medium-v2")
# Example datasets
d4rl_dataset = d4rl_offline_dataset("walker2d-random-v2")
d4rl_dataset = d4rl_offline_dataset("halfcheetah-medium-expert-v2")
d4rl_dataset = d4rl_offline_dataset("antmaze-umaze-v2")
dataset = d4rl_dataset
dataset['observations'].shape, dataset['actions'].shape, dataset['rewards'].shape, dataset['next_observations'].shape, dataset['terminals'].shape
# ((1000000, 11), (1000000, 3), (1000000,), (1000000, 11), (1000000,))
# Create a PyTorch Dataset
d4rl_dataset = d4rl_offline_dataset("hopper-medium-v2")
dataset = D4RLDataset(d4rl_dataset)
# Get a batch of data
batch = dataset[-16:]
batch["observation"].shape, batch["action"].shape, batch["reward"].shape, batch["next_observation"].shape, batch["terminal"].shape
# (torch.Size([16, 11]), torch.Size([16, 3]), torch.Size([16]), torch.Size([16, 11]), torch.Size([16]))
returns = np.random.rand(16, 1)
get_normalized_score = D4RLScoreNormalizer("hopper-medium-v2")
get_normalized_score(returns)
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.4.tar.gz
(9.4 kB
view hashes)
Built Distribution
Close
Hashes for just_d4rl-0.2407.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4bab50d9d723e1f36410a89b64b486113aa6659786b3286a0d2b39dfe20ffbbf |
|
MD5 | f4a12053c7192aa6c43ea6dd7c3f3ec2 |
|
BLAKE2b-256 | 043ca2b57eeaed7696aed89a5e545dd8e8fc7bb018871f91e031007df00ee39b |