Example #1
0
 def _model_fn(self, x, y):
     return models.get_rnn_model(self.rnn_size, self.cell_type,
                                 self.num_layers, self.input_op_fn,
                                 self.bidirectional,
                                 models.linear_regression,
                                 self.sequence_length,
                                 self.initial_state)(x, y)
Example #2
0
 def _model_fn(self, x, y):
   return models.get_rnn_model(self.rnn_size, self.cell_type, self.num_layers,
                               self.input_op_fn, self.bidirectional,
                               models.logistic_regression,
                               self.sequence_length, self.initial_state,
                               self.attn_length, self.attn_size,
                               self.attn_vec_size)(x, y)
Example #3
0
 def _model_fn(self, X, y):
     return models.get_rnn_model(self.rnn_size, self.cell_type,
                                 self.num_layers,
                                 self.input_op_fn, self.bidirectional,
                                 models.linear_regression,
                                 self.sequence_length,
                                 self.initial_state)(X, y)
Example #4
0
 def _model_fn(self, X, y):
     # pylint: disable=invalid-name
     return models.get_rnn_model(self.rnn_size, self.cell_type,
                                 self.num_layers, self.input_op_fn,
                                 self.bidirectional,
                                 models.linear_regression,
                                 self.sequence_length,
                                 self.initial_state)(X, y)
Example #5
0
 def _model_fn(self, X, y):
   # pylint: disable=invalid-name
   return models.get_rnn_model(self.rnn_size, self.cell_type, self.num_layers,
                               self.input_op_fn, self.bidirectional,
                               models.logistic_regression,
                               self.sequence_length, self.initial_state)(X, y)
Example #6
0
 def _model_fn(self, x, y):  # pylint: disable=method-hidden
   return models.get_rnn_model(self.rnn_size, self.cell_type, self.num_layers,
                               self.input_op_fn, self.bidirectional,
                               models.linear_regression, self.sequence_length,
                               self.initial_state, self.attn_length,
                               self.attn_size, self.attn_vec_size)(x, y)
Example #7
0
 def _model_fn(self, x, y):  # pylint: disable=method-hidden
     return models.get_rnn_model(
         self.rnn_size, self.cell_type, self.num_layers, self.input_op_fn,
         self.bidirectional, models.linear_regression, self.sequence_length,
         self.initial_state, self.attn_length, self.attn_size,
         self.attn_vec_size)(x, y)