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.