def getDifference(predicted_class, actual_class): p = SHOT_NAMES.index(predicted_class) a = SHOT_NAMES.index(actual_class) if a == p: return 0 elif abs(a-p) == 1: return 1 else : return 2
def getDataMatrix(file_set, shot=True, leave_out_class=None): """ Returns an Numpy-Array with the feature_lines converted from all the beatscripts mentioned in files. """ matrix = [] classes = [] for file in file_set: feature_lines = getFeatureLinesFromFile(file, shot, leave_out_class=leave_out_class) for line in feature_lines: if shot: classes.append(SHOT_NAMES.index(line.pop())) else: classes.append(int(line.pop()=="True")) matrix.append(np.array(line)) return np.array(matrix, dtype=np.float64), np.array(classes)