d3rlpy.dataset.TransitionPickerProtocol

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

Interface of TransitionPicker.

Methods

__call__(episode, index)[source]

Returns transition specified by index.

Parameters:

  • episode (EpisodeBase) – Episode.

  • index (int) – Index at the target transition.

Returns:

Transition.

Return type:

Transition