d3rlpy.dataset.LastFrameWriterPreprocess

class d3rlpy.dataset.LastFrameWriterPreprocess(*args, **kwargs)

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 (ndarray[Any, dtype[Any]]) – Action.

Returns:

Processed action.

Return type:

ndarray[Any, dtype[Any]]

process_observation(observation)

Processes observation.

Parameters:

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

Returns:

Processed observation.

Return type:

Union[ndarray[Any, dtype[Any]], Sequence[ndarray[Any, dtype[Any]]]]

process_reward(reward)

Processes reward.

Parameters:

reward (ndarray[Any, dtype[Any]]) – Reward.

Returns:

Processed reward.

Return type:

ndarray[Any, dtype[Any]]