d3rlpy.q_functions.MeanQFunctionFactory

class d3rlpy.q_functions.MeanQFunctionFactory

Standard Q function factory class.

This is the standard Q function factory class.

References

• Mnih et al., Human-level control through deep reinforcement learning.
• Lillicrap et al., Continuous control with deep reinforcement learning.

Methods

create(encoder, action_size=None)

Returns PyTorch’s Q function module.

Parameters:

• encoder (torch.nn.Module) – an encoder module that processes the observation (and action in continuous action-space) to obtain feature representations.
• action_size (int) – dimension of discrete action-space. If the action-space is continous, None will be passed.

Returns:

Q function object.

Return type:

torch.nn.Module

get_params(deep=False)

Returns Q function parameters.

Returns:Q function parameters. Return type:dict

get_type()

Returns Q function type.

Returns:Q function type. Return type:str

Attributes

TYPE = 'mean'