d3rlpy.models.QRQFunctionFactory¶
- class d3rlpy.models.QRQFunctionFactory(share_encoder=False, n_quantiles=32)[source]¶
Quantile Regression Q function factory class.
References
- Parameters
- Return type
Methods
- create_continuous(encoder)[source]¶
Returns PyTorch’s Q function module.
- Parameters
encoder (d3rlpy.models.torch.encoders.EncoderWithAction) – an encoder module that processes the observation and action to obtain feature representations.
- Returns
continuous Q function object.
- Return type
d3rlpy.models.torch.q_functions.qr_q_function.ContinuousQRQFunction
- create_discrete(encoder, action_size)[source]¶
Returns PyTorch’s Q function module.
- Parameters
encoder (d3rlpy.models.torch.encoders.Encoder) – an encoder module that processes the observation to obtain feature representations.
action_size (int) – dimension of discrete action-space.
- Returns
discrete Q function object.
- Return type
d3rlpy.models.torch.q_functions.qr_q_function.DiscreteQRQFunction
- classmethod deserialize(serialized_config)¶
- Parameters
serialized_config (str) –
- Return type
d3rlpy.serializable_config.TConfig
- classmethod deserialize_from_dict(dict_config)¶
- Parameters
dict_config (Dict[str, Any]) –
- Return type
d3rlpy.serializable_config.TConfig
- classmethod deserialize_from_file(path)¶
- Parameters
path (str) –
- Return type
d3rlpy.serializable_config.TConfig
- classmethod from_dict(kvs, *, infer_missing=False)¶
- classmethod from_json(s, *, parse_float=None, parse_int=None, parse_constant=None, infer_missing=False, **kw)¶
- classmethod schema(*, infer_missing=False, only=None, exclude=(), many=False, context=None, load_only=(), dump_only=(), partial=False, unknown=None)¶
- to_dict(encode_json=False)¶
- to_json(*, skipkeys=False, ensure_ascii=True, check_circular=True, allow_nan=True, indent=None, separators=None, default=None, sort_keys=False, **kw)¶
Attributes