jat-dataset-tokenized jat-project

Dataset Card for "jat-dataset-tokenized" More Information needed

类型
dataset
下载量
662,576
点赞
23
访问
public
文件
0

标签

  • 机器人数据
  • 强化学习
  • 视频数据集
  • 数据处理
  • 预训练

摘要

该数据集是JAT（Joint Agent Trajectory）项目的tokenized版本，包含大量Atari游戏（如Alien、Asterix、Breakout等）的交互轨迹数据，每个config对应一个游戏。每条样本包含图像观测序列、奖励、离散动作、注意力掩码和损失权重字段，用于训练可处理视觉输入与动作输出的通用agent模型。数据集规模庞大（各游戏训练集约13000-14000条），适用于强化学习、智能体行为建模及决策序列预训练任务。

README

--- dataset_info: - config_name: atari-alien features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: tr…

查看完整页面 · 查看原文