EpisodeData

minari.EpisodeData

class minari.EpisodeData(id: int, seed: int | None, total_timesteps: int, observations: ndarray, actions: ndarray, rewards: ndarray, terminations: ndarray, truncations: ndarray)

Contains the datasets data for a single episode.

This is the object returned by minari.MinariDataset.sample_episodes.

Attributes

EpisodeData.id: int

The ID of the episode in the Minari dataset.

EpisodeData.seed: int | None

The seed used to reset this episode in the Gymnasium API.

EpisodeData.total_timesteps: int

The number of timesteps contained in this episode.

EpisodeData.observations: ndarray

The observations of the environment. The initial and final observations are included meaning that the number of observations will be increased by one compared to the number of timesteps

EpisodeData.actions: ndarray

The actions taken in each episode timestep.

EpisodeData.terminations: ndarray

The terminated value after each environment step.

EpisodeData.truncations: ndarray

The truncated value after each environment step.