예제 #1
0
 def load_target(self, command_index):
     xt = []
     sequence = self.command_data[
         self.command_names[command_index]][0]['target']
     xt.append(
         list(
             parenthesis_split(sequence, self.delimiter, self.lparen,
                               self.rparen)))
     target = pad_to_fixed(xt, self.o_tokens,
                           self.sequence_max_length).squeeze()
     return target
예제 #2
0
 def load_source(self, command_index):
     xs = []
     sequence = self.command_data[
         self.command_names[command_index]][0]['source']
     xs.append(
         list(
             parenthesis_split(sequence, self.delimiter, self.lparen,
                               self.rparen)))
     source = pad_to_fixed(xs, self.i_tokens,
                           self.sequence_max_length).squeeze()
     return source