def get_labels(self, locations, window_size):
        labels = []
        interval = [q for q in range(int(-window_size/2), int(window_size/2) + 1)]
        for loc in locations:
            labels.append([loc + q for q in interval])
        return labels