def apply_padding(self):
params = self.params
self.msgs = padding_message(data=self.msgs,
max_length=params.msg_length)
self.added_code, self.removed_code = clean_and_reformat_code(
self.codes)
self.added_code = padding_commit_code(data=self.added_code,
max_file=params.code_file,
max_line=params.code_line,
max_length=params.code_length)
self.removed_code = padding_commit_code(data=self.removed_code,
max_file=params.code_file,
max_line=params.code_line,
max_length=params.code_length)
cc2ftr.apply_padding()
cc2ftr.apply_mapping()
if params.train is True:
cc2ftr.train()
elif params.predict is True:
data = pickle.load(open(params.predict_data, 'rb'))
ids, labels, msgs, codes = data
dictionary = pickle.load(open(params.dictionary_data, 'rb'))
dict_msg, dict_code = dictionary
pad_msg = padding_message(data=msgs, max_length=params.msg_length)
added_code, removed_code = clean_and_reformat_code(codes)
pad_added_code = padding_commit_code(data=added_code,
max_file=params.code_file,
max_line=params.code_line,
max_length=params.code_length)
pad_removed_code = padding_commit_code(data=removed_code,
max_file=params.code_file,
max_line=params.code_line,
max_length=params.code_length)
pad_msg = mapping_dict_msg(pad_msg=pad_msg, dict_msg=dict_msg)
pad_added_code = mapping_dict_code(pad_code=pad_added_code,
dict_code=dict_code)
pad_removed_code = mapping_dict_code(pad_code=pad_removed_code,
dict_code=dict_code)
pad_msg_labels = convert_msg_to_label(pad_msg=pad_msg,
dict_msg=dict_msg)
data = (pad_added_code, pad_removed_code, pad_msg_labels, dict_msg,