d3rlpy.dataset.BufferProtocol

class d3rlpy.dataset.BufferProtocol(*args, **kwargs)[source]

Interface of Buffer.

Methods

__getitem__(index)[source]
Parameters:

index (int) –

Return type:

Tuple[EpisodeBase, int]

append(episode, index)[source]

Adds transition to buffer.

Parameters:

  • episode (EpisodeBase) – Episode object.

  • index (int) – Transition index.

Return type:

None

Attributes

episodes

Returns list of episodes.

Returns:

List of saved episodes.

transition_count

Returns the number of transitions.

Returns:

Number of transitions.