我正在计算回归问题的加权均方误差。我有你说的是真的,你预言过的,还有你的数字阵列。每个数组都是成形的(N,1),其中N是采样数。我不明白为什么下面两段代码给出了不同的答案:
第一代码段import numpy as np
sq_error = (y_true-y_predicted)**2
wtd_sq_error = np.multiply(sq_error,y_wts)
wtd_mse = np.mean(wtd_sq_error)
第二个代码段取自sklearn metricsmean_squared_error function
^{pr2}$
我来测试这是因为tensorflow加权均方误差和sklearn度量均方误差(指定了权重列)之间的不匹配。请注意,当我不指定权重列时,不会发生这种不匹配。在
谢谢你的帮助!在
因为你忘了重量:
同时
^{pr2}$您的第一个代码段的加权平均值公式错误,应该是:
而不是:
^{pr2}$相关问题 更多 >
编程相关推荐