vote.py 文件源码

def get_segment_vote_accuracy(segment_label, segment_predictions, window):
    def gen():
        count = {
            label: np.hstack([[0], np.cumsum(segment_predictions == label)])
            for label in set(segment_predictions)
        }
        tmp = window
        if tmp == -1:
            tmp = len(segment_predictions)
        tmp = min(tmp, len(segment_predictions))
        for begin in range(len(segment_predictions) - tmp + 1):
            yield segment_label == max(
                count,
                key=lambda label: count[label][begin + tmp] - count[label][begin]
            ), segment_label
    return list(gen())