Python RLPredictor.__init__ Beispiele

Beispiel #1

Datei: discrete_action_predictor.py Projekt: robot-ai-machinelearning/BlueWhale

 def __init__(self, net, parameters, int_features=False):
     RLPredictor.__init__(self, net, parameters, int_features)
     self.is_discrete = True
     self._output_blobs.extend([
         'output/string_single_categorical_features.keys',
         'output/string_single_categorical_features.lengths',
         'output/string_single_categorical_features.values',
     ])

Beispiel #2

Datei: continuous_action_dqn_predictor.py Projekt: wds2014/BlueWhale

 def __init__(self, net, parameters, int_features=False):
     RLPredictor.__init__(self, net, parameters, int_features)
     self.is_discrete = False
     self._output_blobs.extend([
         "output/int_single_categorical_features.keys",
         "output/int_single_categorical_features.lengths",
         "output/int_single_categorical_features.values",
     ])