def prepare_labels(self): labels_pred_ = [] labels_true_ = [] for cp in self.list_of_cp: labels_true_.append(self.map_cp2surgemetransitions[cp]) labels_pred_.append(self.map_cp2cluster[cp]) labels_pred = utils.label_convert_to_numbers(labels_pred_) labels_true = utils.label_convert_to_numbers(labels_true_) assert len(labels_true) == len(labels_pred) assert len(labels_true_) == len(labels_pred_) return labels_true, labels_pred
def prepare_labels(self): labels_pred_1_ = [] labels_pred_2_ = [] labels_true_ = [] for cp in self.list_of_cp: labels_true_.append(self.map_cp2surgemetransitions[cp]) try: milestone_label = self.map_cp2milestones[cp] except KeyError: print "Too Few elements inside cluster!!" sys.exit() labels_pred_1_.append(milestone_label) labels_pred_2_.append(milestone_label.split('_')[0]) labels_pred_1 = utils.label_convert_to_numbers(labels_pred_1_) labels_pred_2 = utils.label_convert_to_numbers(labels_pred_2_) labels_true = utils.label_convert_to_numbers(labels_true_) assert len(labels_true_) == len(labels_pred_1_) == len(labels_pred_2_) assert len(labels_true) == len(labels_pred_1) == len(labels_pred_2) return labels_true, labels_pred_1, labels_pred_2