d3rlpy.dataset.BasicTransitionPicker

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

Standard transition picker.

This class implements a basic transition picking.

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