d3rlpy.dataset.LastFrameWriterPreprocess

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

Data writer that writes the last channel of observation.

This class is designed to be used with FrameStackTransitionPicker.

Methods

process_action(action)

Processes action.

Parameters

action (numpy.ndarray) – Action.

Returns

Processed action.

Return type

numpy.ndarray

process_observation(observation)[source]

Processes observation.

Parameters

observation (Union[numpy.ndarray, Sequence[numpy.ndarray]]) – Observation.

Returns

Processed observation.

Return type

Union[numpy.ndarray, Sequence[numpy.ndarray]]

process_reward(reward)

Processes reward.

Parameters

reward (numpy.ndarray) – Reward.

Returns

Processed reward.

Return type

numpy.ndarray