def cluster_evaluation(self): for cp in self.list_of_cp: demonstration = self.map_cp2demonstrations[cp] frm = self.map_cp2frm[cp] + 2 * self.sr curr_surgeme = self.map_frm2surgeme[demonstration][frm] self.map_cp2surgemes[cp] = curr_surgeme ranges = sorted(utils.get_annotation_segments(constants.PATH_TO_DATA + constants.ANNOTATIONS_FOLDER + demonstration + "_" + constants.CAMERA + ".p")) bin = utils.binary_search(ranges, frm) map_frm2surgeme_demonstration = self.map_frm2surgeme[demonstration] prev_end = bin[0] - 1 next_start = bin[1] + 1 prev_surgeme = map_frm2surgeme_demonstration[prev_end] if prev_end in map_frm2surgeme_demonstration else "start" next_surgeme = map_frm2surgeme_demonstration[next_start] if next_start in map_frm2surgeme_demonstration else "end" surgemetransition = None if abs(frm - (bin[0] - 1)) < abs(bin[1] + 1 - frm): surgemetransition = str(prev_surgeme) + "->" + str(curr_surgeme) else: surgemetransition = str(curr_surgeme) + "->" + str(next_surgeme) self.map_cp2surgemetransitions[cp] = surgemetransition self.cp_surgemes = set(self.map_cp2surgemes.values())
def cluster_evaluation(self): for cp in self.list_of_cp: demonstration = self.map_cp2demonstrations[cp] frm = self.map_cp2frm[cp] curr_surgeme = self.map_frm2surgeme[demonstration][frm] self.map_cp2surgemes[cp] = curr_surgeme ranges = sorted(utils.get_annotation_segments(constants.PATH_TO_DATA + constants.ANNOTATIONS_FOLDER + demonstration + "_" + constants.CAMERA + ".p")) bin = utils.binary_search(ranges, frm) map_frm2surgeme_demonstration = self.map_frm2surgeme[demonstration] prev_end = bin[0] - 1 next_start = bin[1] + 1 prev_surgeme = map_frm2surgeme_demonstration[prev_end] if prev_end in map_frm2surgeme_demonstration else "start" next_surgeme = map_frm2surgeme_demonstration[next_start] if next_start in map_frm2surgeme_demonstration else "end" surgemetransition = None if abs(frm - (bin[0] - 1)) < abs(bin[1] + 1 - frm): surgemetransition = str(prev_surgeme) + "->" + str(curr_surgeme) else: surgemetransition = str(curr_surgeme) + "->" + str(next_surgeme) self.map_cp2surgemetransitions[cp] = surgemetransition self.cp_surgemes = set(self.map_cp2surgemes.values())
def cluster_evaluation(self):
for cp in self.list_of_cp:
demonstration = self.map_cp2demonstrations[cp]
frm = self.map_cp2frm[cp]
curr_surgeme = self.map_frm2surgeme[demonstration][frm]
self.map_cp2surgemes[cp] = curr_surgeme
ranges = sorted(utils.get_annotation_segments(constants.PATH_TO_DATA + constants.ANNOTATIONS_FOLDER
+ demonstration + "_capture2.p"))
bin = utils.binary_search(ranges, frm)
map_frm2surgeme_demonstration = self.map_frm2surgeme[demonstration]
prev_end = bin[0] - 1
next_start = bin[1] + 1
prev_surgeme = map_frm2surgeme_demonstration[prev_end] if prev_end in map_frm2surgeme_demonstration else "start"
next_surgeme = map_frm2surgeme_demonstration[next_start] if next_start in map_frm2surgeme_demonstration else "end"
surgemetransition = None
if abs(frm - (bin[0] - 1)) < abs(bin[1] + 1 - frm):
surgemetransition = str(prev_surgeme) + "->" + str(curr_surgeme)
else:
surgemetransition = str(curr_surgeme) + "->" + str(prev_surgeme)
self.map_cp2surgemetransitions[cp] = surgemetransition
self.cp_surgemes = set(self.map_cp2surgemes.values())
# Initialize data structures
table = {}
for L1_cluster in self.map_level1_cp.keys():
new_dict = {}
for surgeme in self.cp_surgemes:
new_dict[surgeme] = 0
table[L1_cluster] = new_dict
surgeme_count = {}
for surgeme in self.cp_surgemes:
surgeme_count[surgeme] = 0
for L1_cluster in self.map_level1_cp.keys():
list_of_cp_key = self.map_level1_cp[L1_cluster]
for cp in list_of_cp_key:
surgeme = self.map_frm2surgeme[self.map_cp2demonstrations[cp]][self.map_cp2frm[cp]]
surgeme_count[surgeme] += 1
curr_dict = table[L1_cluster]
curr_dict[surgeme] += 1
table[L1_cluster] = curr_dict
final_clusters = list(set(self.map_level1_cp.keys()) - set(self.pruned_L1_clusters))
confusion_matrix = " "
for surgeme in self.cp_surgemes:
confusion_matrix = confusion_matrix + str(surgeme) + " "
# print confusion_matrix
self.file.write('\n\n ---Confusion Matrix--- \n\n')
self.file.write(confusion_matrix)
confusion_matrix = ""
for L1_cluster in final_clusters:
confusion_matrix = confusion_matrix + "\n" + L1_cluster + " "
for surgeme in self.cp_surgemes:
# confusion_matrix += str(float("{0:.2f}".format(table[L1_cluster][surgeme] / float(surgeme_count[surgeme])))) + " "
confusion_matrix += str(round(Decimal(table[L1_cluster][surgeme] / float(surgeme_count[surgeme])), 2)) + " "
confusion_matrix += '\n'
# print confusion_matrix
self.file.write(confusion_matrix)
self.file.write("\n\n ---Surgeme Count--- \n\n")
self.file.write(repr(surgeme_count))
self.file.write("\n\n")
