- Description :
D4RL est une référence open source pour l'apprentissage par renforcement hors ligne. Il fournit des environnements et des ensembles de données standardisés pour les algorithmes de formation et d’analyse comparative.
Les ensembles de données suivent le format RLDS pour représenter les étapes et les épisodes.
Description de la configuration : Voir plus de détails sur la tâche et ses versions sur https://github.com/rail-berkeley/d4rl/wiki/Tasks#adroit
Page d'accueil : https://sites.google.com/view/d4rl-anonymous
Code source :
tfds.d4rl.d4rl_adroit_hammer.D4rlAdroitHammer
Versions :
-
1.0.0
: Version initiale. -
1.1.0
(par défaut) : ajout de is_last.
-
Clés supervisées (Voir doc
as_supervised
) :None
Figure ( tfds.show_examples ) : non pris en charge.
Citation :
@misc{fu2020d4rl,
title={D4RL: Datasets for Deep Data-Driven Reinforcement Learning},
author={Justin Fu and Aviral Kumar and Ofir Nachum and George Tucker and Sergey Levine},
year={2020},
eprint={2004.07219},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
d4rl_adroit_hammer/v0-human (configuration par défaut)
Taille du téléchargement :
5.33 MiB
Taille du jeu de données :
6.10 MiB
Mise en cache automatique ( documentation ) : Oui
Divisions :
Diviser | Exemples |
---|---|
'train' | 70 |
- Structure des fonctionnalités :
FeaturesDict({
'steps': Dataset({
'action': Tensor(shape=(26,), dtype=float32),
'discount': float32,
'infos': FeaturesDict({
'qpos': Tensor(shape=(33,), dtype=float32),
'qvel': Tensor(shape=(33,), dtype=float32),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': Tensor(shape=(46,), dtype=float32),
'reward': float32,
}),
})
- Documentation des fonctionnalités :
Fonctionnalité | Classe | Forme | Type D | Description |
---|---|---|---|---|
FonctionnalitésDict | ||||
pas | Base de données | |||
étapes/actions | Tenseur | (26,) | flotteur32 | |
étapes/remise | Tenseur | flotteur32 | ||
étapes/infos | FonctionnalitésDict | |||
étapes/infos/qpos | Tenseur | (33,) | flotteur32 | |
étapes/infos/qvel | Tenseur | (33,) | flotteur32 | |
étapes/is_first | Tenseur | bouffon | ||
étapes/est_dernier | Tenseur | bouffon | ||
étapes/is_terminal | Tenseur | bouffon | ||
étapes/observation | Tenseur | (46,) | flotteur32 | |
étapes/récompense | Tenseur | flotteur32 |
- Exemples ( tfds.as_dataframe ) :
d4rl_adroit_hammer/v0-cloné
Taille du téléchargement :
644.69 MiB
Taille de l'ensemble de données :
538.97 MiB
Mise en cache automatique ( documentation ) : Non
Divisions :
Diviser | Exemples |
---|---|
'train' | 5 594 |
- Structure des fonctionnalités :
FeaturesDict({
'steps': Dataset({
'action': Tensor(shape=(26,), dtype=float32),
'discount': float64,
'infos': FeaturesDict({
'qpos': Tensor(shape=(33,), dtype=float64),
'qvel': Tensor(shape=(33,), dtype=float64),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': Tensor(shape=(46,), dtype=float64),
'reward': float64,
}),
})
- Documentation des fonctionnalités :
Fonctionnalité | Classe | Forme | Type D | Description |
---|---|---|---|---|
FonctionnalitésDict | ||||
pas | Base de données | |||
étapes/actions | Tenseur | (26,) | flotteur32 | |
étapes/remise | Tenseur | flotteur64 | ||
étapes/infos | FonctionnalitésDict | |||
étapes/infos/qpos | Tenseur | (33,) | flotteur64 | |
étapes/infos/qvel | Tenseur | (33,) | flotteur64 | |
étapes/is_first | Tenseur | bouffon | ||
étapes/est_dernier | Tenseur | bouffon | ||
étapes/is_terminal | Tenseur | bouffon | ||
étapes/observation | Tenseur | (46,) | flotteur64 | |
étapes/récompense | Tenseur | flotteur64 |
- Exemples ( tfds.as_dataframe ) :
d4rl_adroit_hammer/v0-expert
Taille du téléchargement :
529.91 MiB
Taille de l'ensemble de données :
737.00 MiB
Mise en cache automatique ( documentation ) : Non
Divisions :
Diviser | Exemples |
---|---|
'train' | 5 000 |
- Structure des fonctionnalités :
FeaturesDict({
'steps': Dataset({
'action': Tensor(shape=(26,), dtype=float32),
'discount': float32,
'infos': FeaturesDict({
'action_logstd': Tensor(shape=(26,), dtype=float32),
'action_mean': Tensor(shape=(26,), dtype=float32),
'qpos': Tensor(shape=(33,), dtype=float32),
'qvel': Tensor(shape=(33,), dtype=float32),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': Tensor(shape=(46,), dtype=float32),
'reward': float32,
}),
})
- Documentation des fonctionnalités :
Fonctionnalité | Classe | Forme | Type D | Description |
---|---|---|---|---|
FonctionnalitésDict | ||||
pas | Base de données | |||
étapes/actions | Tenseur | (26,) | flotteur32 | |
étapes/remise | Tenseur | flotteur32 | ||
étapes/infos | FonctionnalitésDict | |||
étapes/infos/action_logstd | Tenseur | (26,) | flotteur32 | |
étapes/infos/action_mean | Tenseur | (26,) | flotteur32 | |
étapes/infos/qpos | Tenseur | (33,) | flotteur32 | |
étapes/infos/qvel | Tenseur | (33,) | flotteur32 | |
étapes/is_first | Tenseur | bouffon | ||
étapes/est_dernier | Tenseur | bouffon | ||
étapes/is_terminal | Tenseur | bouffon | ||
étapes/observation | Tenseur | (46,) | flotteur32 | |
étapes/récompense | Tenseur | flotteur32 |
- Exemples ( tfds.as_dataframe ) :
d4rl_adroit_hammer/v1-humain
Taille du téléchargement :
5.35 MiB
Taille de l'ensemble de données :
6.34 MiB
Mise en cache automatique ( documentation ) : Oui
Divisions :
Diviser | Exemples |
---|---|
'train' | 25 |
- Structure des fonctionnalités :
FeaturesDict({
'steps': Dataset({
'action': Tensor(shape=(26,), dtype=float32),
'discount': float32,
'infos': FeaturesDict({
'board_pos': Tensor(shape=(3,), dtype=float32),
'qpos': Tensor(shape=(33,), dtype=float32),
'qvel': Tensor(shape=(33,), dtype=float32),
'target_pos': Tensor(shape=(3,), dtype=float32),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': Tensor(shape=(46,), dtype=float32),
'reward': float32,
}),
})
- Documentation des fonctionnalités :
Fonctionnalité | Classe | Forme | Type D | Description |
---|---|---|---|---|
FonctionnalitésDict | ||||
pas | Base de données | |||
étapes/actions | Tenseur | (26,) | flotteur32 | |
étapes/remise | Tenseur | flotteur32 | ||
étapes/infos | FonctionnalitésDict | |||
étapes/infos/board_pos | Tenseur | (3,) | flotteur32 | |
étapes/infos/qpos | Tenseur | (33,) | flotteur32 | |
étapes/infos/qvel | Tenseur | (33,) | flotteur32 | |
étapes/infos/target_pos | Tenseur | (3,) | flotteur32 | |
étapes/is_first | Tenseur | bouffon | ||
étapes/est_dernier | Tenseur | bouffon | ||
étapes/is_terminal | Tenseur | bouffon | ||
étapes/observation | Tenseur | (46,) | flotteur32 | |
étapes/récompense | Tenseur | flotteur32 |
- Exemples ( tfds.as_dataframe ) :
d4rl_adroit_hammer/v1-cloné
Taille du téléchargement :
425.93 MiB
Taille du jeu de données :
1.68 GiB
Mise en cache automatique ( documentation ) : Non
Divisions :
Diviser | Exemples |
---|---|
'train' | 3 606 |
- Structure des fonctionnalités :
FeaturesDict({
'algorithm': string,
'policy': FeaturesDict({
'fc0': FeaturesDict({
'bias': Tensor(shape=(256,), dtype=float32),
'weight': Tensor(shape=(46, 256), dtype=float32),
}),
'fc1': FeaturesDict({
'bias': Tensor(shape=(256,), dtype=float32),
'weight': Tensor(shape=(256, 256), dtype=float32),
}),
'last_fc': FeaturesDict({
'bias': Tensor(shape=(26,), dtype=float32),
'weight': Tensor(shape=(256, 26), dtype=float32),
}),
'nonlinearity': string,
'output_distribution': string,
}),
'steps': Dataset({
'action': Tensor(shape=(26,), dtype=float32),
'discount': float32,
'infos': FeaturesDict({
'board_pos': Tensor(shape=(3,), dtype=float32),
'qpos': Tensor(shape=(33,), dtype=float32),
'qvel': Tensor(shape=(33,), dtype=float32),
'target_pos': Tensor(shape=(3,), dtype=float32),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': Tensor(shape=(46,), dtype=float32),
'reward': float32,
}),
})
- Documentation des fonctionnalités :
Fonctionnalité | Classe | Forme | Type D | Description |
---|---|---|---|---|
FonctionnalitésDict | ||||
algorithme | Tenseur | chaîne | ||
politique | FonctionnalitésDict | |||
politique/fc0 | FonctionnalitésDict | |||
politique/fc0/biais | Tenseur | (256,) | flotteur32 | |
politique/fc0/poids | Tenseur | (46, 256) | flotteur32 | |
politique/fc1 | FonctionnalitésDict | |||
politique/fc1/biais | Tenseur | (256,) | flotteur32 | |
politique/fc1/poids | Tenseur | (256, 256) | flotteur32 | |
politique/last_fc | FonctionnalitésDict | |||
politique/last_fc/bias | Tenseur | (26,) | flotteur32 | |
politique/last_fc/poids | Tenseur | (256, 26) | flotteur32 | |
politique/non-linéarité | Tenseur | chaîne | ||
politique/distribution_de sortie | Tenseur | chaîne | ||
pas | Base de données | |||
étapes/actions | Tenseur | (26,) | flotteur32 | |
étapes/remise | Tenseur | flotteur32 | ||
étapes/infos | FonctionnalitésDict | |||
étapes/infos/board_pos | Tenseur | (3,) | flotteur32 | |
étapes/infos/qpos | Tenseur | (33,) | flotteur32 | |
étapes/infos/qvel | Tenseur | (33,) | flotteur32 | |
étapes/infos/target_pos | Tenseur | (3,) | flotteur32 | |
étapes/is_first | Tenseur | bouffon | ||
étapes/est_dernier | Tenseur | bouffon | ||
étapes/is_terminal | Tenseur | bouffon | ||
étapes/observation | Tenseur | (46,) | flotteur32 | |
étapes/récompense | Tenseur | flotteur32 |
- Exemples ( tfds.as_dataframe ) :
d4rl_adroit_hammer/v1-expert
Taille du téléchargement :
531.24 MiB
Taille de l'ensemble de données :
843.54 MiB
Mise en cache automatique ( documentation ) : Non
Divisions :
Diviser | Exemples |
---|---|
'train' | 5 000 |
- Structure des fonctionnalités :
FeaturesDict({
'algorithm': string,
'policy': FeaturesDict({
'fc0': FeaturesDict({
'bias': Tensor(shape=(32,), dtype=float32),
'weight': Tensor(shape=(32, 46), dtype=float32),
}),
'fc1': FeaturesDict({
'bias': Tensor(shape=(32,), dtype=float32),
'weight': Tensor(shape=(32, 32), dtype=float32),
}),
'last_fc': FeaturesDict({
'bias': Tensor(shape=(26,), dtype=float32),
'weight': Tensor(shape=(26, 32), dtype=float32),
}),
'last_fc_log_std': FeaturesDict({
'bias': Tensor(shape=(26,), dtype=float32),
'weight': Tensor(shape=(26, 32), dtype=float32),
}),
'nonlinearity': string,
'output_distribution': string,
}),
'steps': Dataset({
'action': Tensor(shape=(26,), dtype=float32),
'discount': float32,
'infos': FeaturesDict({
'action_log_std': Tensor(shape=(26,), dtype=float32),
'action_mean': Tensor(shape=(26,), dtype=float32),
'board_pos': Tensor(shape=(3,), dtype=float32),
'qpos': Tensor(shape=(33,), dtype=float32),
'qvel': Tensor(shape=(33,), dtype=float32),
'target_pos': Tensor(shape=(3,), dtype=float32),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': Tensor(shape=(46,), dtype=float32),
'reward': float32,
}),
})
- Documentation des fonctionnalités :
Fonctionnalité | Classe | Forme | Type D | Description |
---|---|---|---|---|
FonctionnalitésDict | ||||
algorithme | Tenseur | chaîne | ||
politique | FonctionnalitésDict | |||
politique/fc0 | FonctionnalitésDict | |||
politique/fc0/biais | Tenseur | (32,) | flotteur32 | |
politique/fc0/poids | Tenseur | (32, 46) | flotteur32 | |
politique/fc1 | FonctionnalitésDict | |||
politique/fc1/biais | Tenseur | (32,) | flotteur32 | |
politique/fc1/poids | Tenseur | (32, 32) | flotteur32 | |
politique/last_fc | FonctionnalitésDict | |||
politique/last_fc/bias | Tenseur | (26,) | flotteur32 | |
politique/last_fc/poids | Tenseur | (26, 32) | flotteur32 | |
politique/last_fc_log_std | FonctionnalitésDict | |||
politique/last_fc_log_std/bias | Tenseur | (26,) | flotteur32 | |
politique/last_fc_log_std/poids | Tenseur | (26, 32) | flotteur32 | |
politique/non-linéarité | Tenseur | chaîne | ||
politique/distribution_de sortie | Tenseur | chaîne | ||
pas | Base de données | |||
étapes/actions | Tenseur | (26,) | flotteur32 | |
étapes/remise | Tenseur | flotteur32 | ||
étapes/infos | FonctionnalitésDict | |||
étapes/infos/action_log_std | Tenseur | (26,) | flotteur32 | |
étapes/infos/action_mean | Tenseur | (26,) | flotteur32 | |
étapes/infos/board_pos | Tenseur | (3,) | flotteur32 | |
étapes/infos/qpos | Tenseur | (33,) | flotteur32 | |
étapes/infos/qvel | Tenseur | (33,) | flotteur32 | |
étapes/infos/target_pos | Tenseur | (3,) | flotteur32 | |
étapes/is_first | Tenseur | bouffon | ||
étapes/est_dernier | Tenseur | bouffon | ||
étapes/is_terminal | Tenseur | bouffon | ||
étapes/observation | Tenseur | (46,) | flotteur32 | |
étapes/récompense | Tenseur | flotteur32 |
- Exemples ( tfds.as_dataframe ) :