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, hidden_size)[source]¶
Returns PyTorch’s Q function module.
- Parameters
encoder (d3rlpy.models.torch.encoders.EncoderWithAction) – Encoder module that processes the observation and action to obtain feature representations.
hidden_size (int) – Dimension of encoder output.
- Returns
Tuple of continuous Q function and its forwarder.
- Return type
Tuple[d3rlpy.models.torch.q_functions.qr_q_function.ContinuousQRQFunction, d3rlpy.models.torch.q_functions.qr_q_function.ContinuousQRQFunctionForwarder]
- create_discrete(encoder, hidden_size, action_size)[source]¶
Returns PyTorch’s Q function module.
- Parameters
- Returns
Tuple of discrete Q function and its forwarder.
- Return type
Tuple[d3rlpy.models.torch.q_functions.qr_q_function.DiscreteQRQFunction, d3rlpy.models.torch.q_functions.qr_q_function.DiscreteQRQFunctionForwarder]
- 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