d3rlpy.dataset.TransitionPickerProtocol

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

Interface of TransitionPicker.

Methods

__call__(episode, index)[source]

Returns transition specified by index.

Parameters

  • episode (d3rlpy.dataset.components.EpisodeBase) – Episode.

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

Returns

Transition.

Return type

d3rlpy.dataset.components.Transition