This repository contains the Dark Key-Door datasets as used in "Retrieval-augmented Decision Transformer: External Memory for In-context RL":
Datasets for 3 grid-sizes: 10x10, 20x20, 40x20.
Every split contains 100 .npz files, corresponing to 100 start-goal position combinations.
Each file contains 100K transitions.
Filenames have the format START_KEY_DOOR.npz. E.g., for [0,0]
[0,5][0,7].npz, [0.0] is the start position, the key is located at [0,5], and the door is… See the full description on the dataset page:
https://huggingface.co/datasets/ml-jku/dark_keydoor.
