def predictor(self) -> ParametricDQNPredictor:
"""Builds a ParametricDQNPredictor."""
return ParametricDQNPredictor.export(
self,
self.state_normalization_parameters,
self.action_normalization_parameters,
self.use_gpu,
)
def predictor(self) -> ParametricDQNPredictor:
"""Builds a ParametricDQNPredictor."""
return ParametricDQNPredictor.export(
self,
self.state_normalization_parameters,
self.action_normalization_parameters,
self._additional_feature_types.int_features,
self.use_gpu,
)
def predictor(self) -> ParametricDQNPredictor:
"""Builds a ParametricDQNPredictor."""
return ParametricDQNPredictor.export(
self,
self.state_normalization_parameters,
self.action_normalization_parameters,
self._additional_feature_types.int_features,
self.use_gpu,
)
def export_critic(
cls,
trainer,
state_normalization_parameters,
action_normalization_parameters,
model_on_gpu=False,
):
return ParametricDQNPredictor.export(
trainer,
state_normalization_parameters,
action_normalization_parameters,
model_on_gpu=model_on_gpu,
normalize_actions=False,
)
def export_critic(
cls,
trainer,
state_normalization_parameters,
action_normalization_parameters,
int_features=False,
model_on_gpu=False,
):
return ParametricDQNPredictor.export(
trainer,
state_normalization_parameters,
action_normalization_parameters,
int_features=int_features,
model_on_gpu=model_on_gpu,
normalize_actions=False,
)