sklearn.utils.extmath.weighted_mode¶
sklearn.utils.extmath.weighted_mode(a, w, *, axis=0)
返回数组中加权模态(最常见)值的数组
如果有多个这样的值,则仅返回第一个。 还将返回模态箱的箱数。
这是scipy.stats.mode中算法的扩展。
参数 | 说明 |
---|---|
a | array_like 查找模式的n维数组。 |
w | array_like 每个值的权重的n维数组 |
axis | int, optional 操作的轴。默认值为0,即第一个轴。 |
返回值 | 说明 |
---|---|
vals | ndarray 模态值数组。 |
score | ndarray 每种模式的加权计数数组。 |
另见:
示例:
>>> from sklearn.utils.extmath import weighted_mode
>>> x = [4, 1, 4, 2, 4, 2]
>>> weights = [1, 1, 1, 1, 1, 1]
>>> weighted_mode(x, weights)
(array([4.]), array([3.]))
值4出现了三次:在权重一致的情况下,结果就是分布的模式。
>>> weights = [1, 3, 0.5, 1.5, 1, 2] # deweight the 4's
>>> weighted_mode(x, weights)
(array([2.]), array([3.5]))
值2得分最高:权重为1.5和2时,它出现两次:权重之和为3.5。