How to get observation_next

takuseno / d4rl-atari

Datasets for data-driven deep reinforcement learning with Atari (wrapper for datasets released by Google)

MIT License

101 stars 14 forks source link

Closed j0rd1smit closed 3 years ago

j0rd1smit commented 3 years ago

How do I get the next observations? Can I assume that the dataset is ordered? If so the following should work, right?

observation = dataset['observations'][i]
observation_next = dataset['observations'][i + 1]

takuseno commented 3 years ago

@j0rd1smit Hello. Yes, the data is chronologically ordered.

takuseno commented 3 years ago

Let me close this issue since this seems to be resolved.