d4rl_antmaze

  • Mô tả :

D4RL là một điểm chuẩn mã nguồn mở cho việc học tăng cường ngoại tuyến. Nó cung cấp các môi trường và bộ dữ liệu được tiêu chuẩn hóa cho các thuật toán đào tạo và điểm chuẩn.

Bộ dữ liệu tuân theo định dạng RLDS để biểu thị các bước và các tập.

FeaturesDict({
    'steps': Dataset({
        'action': Tensor(shape=(8,), dtype=tf.float32),
        'discount': tf.float32,
        'infos': FeaturesDict({
            'goal': Tensor(shape=(2,), dtype=tf.float32),
            'qpos': Tensor(shape=(15,), dtype=tf.float32),
            'qvel': Tensor(shape=(14,), dtype=tf.float32),
        }),
        'is_first': tf.bool,
        'is_last': tf.bool,
        'is_terminal': tf.bool,
        'observation': Tensor(shape=(29,), dtype=tf.float32),
        'reward': tf.float32,
    }),
})
  • Tài liệu tính năng :
Tính năng Lớp Hình dạng Loại Sự miêu tả
Các tính năng
các bước Dataset
các bước / hành động Tensor (số 8,) tf.float32
bước / chiết khấu Tensor tf.float32
các bước / thông tin Các tính năng
bước / thông tin / mục tiêu Tensor (2,) tf.float32
step / infos / qpos Tensor (15,) tf.float32
step / infos / qvel Tensor (14,) tf.float32
bước / is_first Tensor tf.bool
bước / is_last Tensor tf.bool
step / is_terminal Tensor tf.bool
các bước / quan sát Tensor (29,) tf.float32
bước / phần thưởng Tensor tf.float32
@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_antmaze / umaze-v0 (cấu hình mặc định)

  • Kích thước tải xuống : 221.76 MiB

  • Kích thước tập dữ liệu: 274.83 MiB

  • Tách :

Tách ra Các ví dụ
'train' 10.154

d4rl_antmaze / umaze-đa dạng-v0

  • Kích thước tải xuống : 220.16 MiB

  • Kích thước tập dữ liệu: 270.23 MiB

  • Tách :

Tách ra Các ví dụ
'train' 1.154

d4rl_antmaze / medium-play-v0

  • Kích thước tải xuống : 220.40 MiB

  • Kích thước tập dữ liệu: 275.20 MiB

  • Tách :

Tách ra Các ví dụ
'train' 10.695

d4rl_antmaze / medium-đa dạng-v0

  • Kích thước tải xuống : 220.39 MiB

  • Kích thước tập dữ liệu: 271.18 MiB

  • Tách :

Tách ra Các ví dụ
'train' 2.924

d4rl_antmaze / Large-đa dạng-v0

  • Kích thước tải xuống : 220.47 MiB

  • Kích thước tập dữ liệu: 273.36 MiB

  • Tách :

Tách ra Các ví dụ
'train' 7.141

d4rl_antmaze / large-play-v0

  • Kích thước tải xuống : 220.19 MiB

  • Kích thước tập dữ liệu: 276.61 MiB

  • Tách :

Tách ra Các ví dụ
'train' 13.458