EpisodeData¶
minari.EpisodeData¶
- class minari.EpisodeData(id: int, observations: Any, actions: Any, rewards: ndarray, terminations: ndarray, truncations: ndarray, infos: dict)[source]¶
Contains the datasets data for a single episode.
Attributes¶
- EpisodeData.id: int¶
The ID of the episode in the Minari dataset.
- EpisodeData.observations: Any¶
Stacked observations of the episode. The initial and final observations are included meaning that the number of observations will be increased by one compared to the number of steps.
- EpisodeData.actions: Any¶
Stacked actions taken in each episode step.
- EpisodeData.terminations: ndarray¶
The
terminated
value after each environment step.
- EpisodeData.truncations: ndarray¶
The
truncated
value after each environment step.
- EpisodeData.infos: dict¶
The stacked
infos
of the episodes. As for the observations, this attribute contains one more element compared to the number of steps.