kyutai/rocket-science

O dataset kyutai/rocket-science está entre os destaques do Hugging Face — dados que alimentam o treinamento e a avaliação dos modelos do momento.

Ficha do dataset

Sobre o dataset

Rocket Science Time-aligned video, keyboard actions, game events, and per-frame game state for all four players, captured from a 2v2 Rocket League match.

Como carregar

Use a biblioteca datasets do Hugging Face:

pip install -U datasets

Como é um dataset grande, vale carregar em modo streaming (sem baixar tudo):

from datasets import load_dataset

ds = load_dataset("kyutai/rocket-science", split="train", streaming=True)
for exemplo in ds.take(3):
    print(exemplo)

Tags

other rocket-league world-model video reinforcement-learning multimodal game webdataset

Explorar o dataset no Hugging Face →