qgallouedec/prj_gia_dataset_metaworld_assembly_v2_1111
收藏资源简介:
--- library_name: gia tags: - deep-reinforcement-learning - reinforcement-learning - gia - multi-task - multi-modal - imitation-learning - offline-reinforcement-learning --- An imitation learning environment for the assembly-v2 environment, sample for the policy assembly-v2 This environment was created as part of the Generally Intelligent Agents project gia: https://github.com/huggingface/gia ## Load dataset First, clone it with ```sh git clone https://huggingface.co/datasets/qgallouedec/prj_gia_dataset_metaworld_assembly_v2_1111 ``` Then, load it with ```python import numpy as np dataset = np.load("prj_gia_dataset_metaworld_assembly_v2_1111/dataset.npy", allow_pickle=True).item() print(dataset.keys()) # dict_keys(['observations', 'actions', 'dones', 'rewards']) ```
数据集概述
数据集名称
- 名称: prj_gia_dataset_metaworld_assembly_v2_1111
数据集标签
- 标签:
- deep-reinforcement-learning
- reinforcement-learning
- gia
- multi-task
- multi-modal
- imitation-learning
- offline-reinforcement-learning
数据集描述
- 描述: 这是一个针对assembly-v2环境的模仿学习环境,用于演示policy assembly-v2。
数据集加载
- 加载方法:
-
首先,通过以下命令克隆数据集: sh git clone https://huggingface.co/datasets/qgallouedec/prj_gia_dataset_metaworld_assembly_v2_1111
-
然后,使用以下Python代码加载数据集: python import numpy as np dataset = np.load("prj_gia_dataset_metaworld_assembly_v2_1111/dataset.npy", allow_pickle=True).item() print(dataset.keys()) # 输出: dict_keys([observations, actions, dones, rewards])
-



