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)
