d3rlpy.dataset.TransitionMiniBatch

class d3rlpy.dataset.TransitionMiniBatch(transitions)[source]

mini-batch of Transition objects.

This class is designed to hold d3rlpy.dataset.Transition objects for being passed to algorithms during fitting.

Parameters:transitions (list(d3rlpy.dataset.Transition)) – mini-batch of transitions.

Methods

__getitem__(index)[source]
__len__()[source]
__iter__()[source]
size()[source]

Returns size of mini-batch.

Returns:mini-batch size.
Return type:int

Attributes

actions

Returns mini-batch of actions at t.

Returns:actions at t.
Return type:numpy.ndarray
next_actions

Returns mini-batch of actions at t+1.

Returns:actions at t+1.
Return type:numpy.ndarray
next_observations

Returns mini-batch of observations at t+1.

Returns:observations at t+1.
Return type:numpy.ndarray
next_rewards

Returns mini-batch of rewards at t+1.

Returns:rewards at t+1.
Return type:numpy.ndarray
observations

Returns mini-batch of observations at t.

Returns:observations at t.
Return type:numpy.ndarray
rewards

Returns mini-batch of rewards at t.

Returns:rewards at t.
Return type:numpy.ndarray
terminals

Returns mini-batch of terminal flags at t+1.

Returns:terminal flags at t+1.
Return type:numpy.ndarray
transitions

Returns transitions.

Returns:list of transitions.
Return type:d3rlpy.dataset.Transition