INNER CODE UNIT · Python

ensemble

yuhaozhang/tacred-relation · ensemble.py:48

def ensemble(scores):
    """
    Ensemble by majority vote.
    """
    c = Counter()
    for probs in zip(scores):
        idx = int(np.argmax(np.array(probs)))
        c.update([idx])
    best = c.most_common(1)[0][0]
    return best

if __name__ == '__main__':
    main()

View source record →