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
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