繁体   English   中英

sklearn 给出了意外的 r2 分数

[英]sklearn gives unexpected r2 score

我已经建立了一个linear regression模型,我想根据我的输出计算r2分数。 然而结果真的出乎意料:

正如您在下面看到的, yy hat之间的pearson相关性是正的,这意味着r2分数至少应该是正的。 但是我从sklearn得到的结果是否定的。 怎么来的? 提前致谢!


import numpy as np
from sklearn.metrics import r2_score
from scipy.stats import pearsonr

y = np.array([ 5.2       ,  1.144     ,  3.3       ,  5.59741373,  1.438     ,       7.562     ,  2.7       ,  0.22706035,  2.204     ,  2.396     ,
        4.314     , 12.51420331, 10.8       , 10.638     ,  5.101     ,
        3.861     ,  3.2       ,  3.8       ,  7.072     , -0.4597798 ,
       -0.9       ,  0.3       , -3.54      , -0.4       , -3.        ,
        0.7       ,  1.3       ,  1.5       ,  6.        ,  2.8       ,
        2.        ,  3.122     ])

y_hat = np.array([ 1.25131326,  2.64864629,  1.56201996,  4.26699994,  2.21499358,
        0.59113701,  2.40848854,  0.14954989,  0.45800824,  2.82399621,
        2.48736001,  2.78476975,  1.36378354,  3.4889863 ,  2.4226333 ,
        2.63939523,  4.15008518,  2.61525276,  2.29859288, -1.4358969 ,
       -3.67752652, -3.73173215, -2.67027158,  0.35012302,  3.91349371,
        5.11971861,  5.96586311,  3.36520449,  0.5204047 ,  1.584193  ,
       -0.05781178,  1.75957967])

pearsonr(y, y_hat) # This gives around 0.299
r2_score(y, y_hat) # This gives -0.18478241562914666

我想我知道这里发生了什么。 基本上我天真地认为正相关会导致正 r 平方,但事实并非如此。 通过计算 y_hat 与 y 和 y_avg 与 y 的均方误差,我意识到 y_hat 确实是比总是只预测平均值更糟糕的估计量。

http://www.fairlynerdy.com/what-is-r-squared/

从上面的链接看这张图,你可以看到,即使两个系列在同一个方向上移动,截距造成的距离也会让 MSE 测量的性能非常糟糕在此处输入图片说明

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM