scikit-learn - 模型 score() 与 r2_score 之间的差异

Question

我正在训练一个 LinearRegression() 分类器并试图衡量它的预测准确性

from sklearn.metrics import r2_score
from sklearn.linear_model import LinearRegression
regr_rf = LinearRegression()

regr_rf.fit(df[features],df['label'])
y_rf = regr_rf.predict(df[features])
score = regr_rf.score(df[features],df['label'])
print score
score2 = r2_score(y_rf,df['label'])
print score2

score 和 score2 都显示出非常不同的值。我虽然假设模型的得分函数与显式计算的 r2_score 相同

Answer 1

您对 r2_score 的使用是错误的。第一个参数应该是真实值，而不是预测值。

根据 the documentation :

r2_score(y_true, y_pred, ...)

所以改变这一行 score2 = r2_score(y_rf,df['label'])在您的代码中:

score2 = r2_score(df['label'], y_rf)

然后比较结果。