if labels is None:
labels = annotation.labels()
indices = {label: i for i, label in enumerate(labels)}
// number of samples
N = len(features)
After Change
if labels is None:
labels = [get_label_identifier(label, current_file)
for label in annotation.labels()]
indices = {label: i for i, label in enumerate(labels)}
// number of samples