コード例 #1
0
 def predictor(self) -> DQNPredictor:
     """Builds a DQNPredictor."""
     return DQNPredictor.export(
         self,
         self._actions,
         self.state_normalization_parameters,
         self._additional_feature_types.int_features,
         self.use_gpu,
     )
コード例 #2
0
ファイル: dqn_trainer.py プロジェクト: giang12/Horizon
 def predictor(self, set_missing_value_to_zero=False) -> DQNPredictor:
     """Builds a DQNPredictor."""
     return DQNPredictor.export(
         self,
         self._actions,
         self.state_normalization_parameters,
         self.use_gpu,
         set_missing_value_to_zero=set_missing_value_to_zero,
     )