Example #1
0
 def create_input_layer(self, input_placeholder: BaseInputFormatter):
     """Creates keras model"""
     if self.prediction_mode:
         shape = [1] + input_placeholder.get_input_state_dimension()
         self.inputs = Input(batch_input_shape=shape)
     else:
         self.inputs = Input(shape=input_placeholder.get_input_state_dimension())
     return self.inputs
Example #2
0
 def create_input_layer(self, input_placeholder: BaseInputFormatter):
     """Creates keras model"""
     model = tf.keras.Sequential()
     model.add(
         tf.keras.layers.InputLayer(
             input_shape=input_placeholder.get_input_state_dimension()))
     self.model = model
Example #3
0
 def create_input_layer(self, input_placeholder: BaseInputFormatter):
     """Creates keras model"""
     self.inputs = tf.keras.layers.InputLayer(
         input_shape=input_placeholder.get_input_state_dimension())
     return self.inputs
Example #4
0
 def create_input_layer(self, input_placeholder: BaseInputFormatter):
     self.input_dim = input_placeholder.get_input_state_dimension()[-1]
     self.inputs = tf.keras.layers.Input(shape=(self.input_dim, ))