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
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
-