Bases: Aggregate_Base
Aggregate function based on majority vote.
This class should be passed to an ensemble function/class for combining predictions.
Source code in aucmedi/ensemble/aggregate/majority_vote.py
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54 | class MajorityVote(Aggregate_Base):
""" Aggregate function based on majority vote.
This class should be passed to an ensemble function/class for combining predictions.
"""
#---------------------------------------------#
# Initialization #
#---------------------------------------------#
def __init__(self):
# No hyperparameter adjustment required for this method, therefore skip
pass
#---------------------------------------------#
# Aggregate #
#---------------------------------------------#
def aggregate(self, preds):
# Count votes
votes = np.argmax(preds, axis=1)
# Identify majority
majority_vote = np.argmax(np.bincount(votes))
# Create prediction based on majority vote
pred = np.zeros((preds.shape[1]))
pred[majority_vote] = 1
# Return prediction
return pred
|