def create(backbone: ModelFactory,
input_block: typing.Optional[ModelFactory] = None):
""" Vel factory function """
if input_block is None:
input_block = IdentityFactory()
return QModelFactory(input_block=input_block, backbone=backbone)
def create(policy_backbone: ModelFactory,
value_backbone: ModelFactory,
input_block: typing.Optional[ModelFactory] = None):
""" Vel factory function """
if input_block is None:
input_block = IdentityFactory()
return DeterministicPolicyModelFactory(input_block=input_block,
policy_backbone=policy_backbone,
value_backbone=value_backbone)
def create(backbone: ModelFactory,
input_block: typing.Optional[ModelFactory] = None,
initial_std_dev=0.4,
factorized_noise=True):
""" Vel factory function """
if input_block is None:
input_block = IdentityFactory()
return NoisyQModelFactory(input_block=input_block,
backbone=backbone,
initial_std_dev=initial_std_dev,
factorized_noise=factorized_noise)
def create(backbone: ModelFactory,
vmin: float,
vmax: float,
atoms: int,
input_block: typing.Optional[ModelFactory] = None):
""" Vel factory function """
if input_block is None:
input_block = IdentityFactory()
return QDistributionalModelFactory(input_block=input_block,
backbone=backbone,
vmin=vmin,
vmax=vmax,
atoms=atoms)
def create(backbone: ModelFactory,
vmin: float,
vmax: float,
atoms: int,
initial_std_dev: float = 0.4,
factorized_noise: bool = True,
input_block: typing.Optional[ModelFactory] = None):
""" Vel factory function """
if input_block is None:
input_block = IdentityFactory()
return QDistributionalModelFactory(input_block=input_block,
backbone=backbone,
vmin=vmin,
vmax=vmax,
atoms=atoms,
initial_std_dev=initial_std_dev,
factorized_noise=factorized_noise)