我正在使用 sklearn 构建线性回归模型和决策树模型。我想比较这两个模型的性能，我已经计算了两个模型的 r2_score。我已经计算了这两个值的 model.score。我很困惑哪个是比较这些模型性能的更好指标。还有什么 model.score 给出的？
from sklearn.metrics import r2_score
score_DT = r2_score(y_pred_DT,y_test)
dt_score = regressorDT.score(X_test,y_test)
model.score 和 r2_score 为回归模型提供不同的值
数据挖掘
机器学习
scikit-学习
回归
决策树
线性回归
2021-09-21 03:33:59
1个回答
这两个函数是相同的 r2 度量并且应该产生相同的结果。
您对 r2_score 函数的使用是错误的。第一个参数应该是基本真实值而不是预测值，因此在您的情况下应该是：
score_DT = r2_score(y_test, y_pred_DT)
其它你可能感兴趣的问题