首页 > 解决方案 > 如果使用votingclassifier() 进行集成时投票相等怎么办?

问题描述

例如,当 4 个模型预测两个独特的结果时会发生什么:1、1、0 和 0。votingclassifier()当没有明确的多数时该怎么办。

标签: pythonmachine-learningscikit-learnensemble-learning

解决方案


从 sci-kit 学习文档VotingClassifier.predict()

https://github.com/scikit-learn/scikit-learn/blob/1495f6924/sklearn/ensemble/voting.py#L281

我们可以看到这个函数调用numpy.argmax

从它的文档中,在“注释”下,它说“如果多次出现最大值,则返回与第一次出现相对应的索引。” https://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html

因此,在您的示例中,它将预测分类列表中首先出现的那个1

如果 VotingClassifier 使用“软”投票,并且两个结果的概率和相同,它将预测结果列表中的第一个结果


推荐阅读