Python R-Squared: 计算线性回归模型拟合度的评估指标

Python 笔记

一、什么是R-Squared？

R-Squared指标是用于衡量回归方程的拟合程度的评估指标，它表示拟合线与实际数据点的总偏离程度的平方和与总变差程度的平方和的比值。R-Squared值越接近1，表示模型的拟合程度越好，接近0则表示模型的拟合程度越差。

在Python中，我们可以使用相关库来计算R-Squared指标，如：

import numpy as np 
from sklearn.metrics import r2_score

y_true = np.array([1, 2, 3])
y_pred = np.array([1, 2.5, 3])
r2_score(y_true, y_pred)

上述代码中，我们使用了numpy和sklearn库，分别导入r2_score函数和numpy库。我们传入了真实值数组y_true和预测值数组y_pred，并调用r2_score函数进行计算，返回R-Squared值。

二、计算R-Squared的步骤

计算R-Squared指标的步骤可以分为以下几步：

1. 计算总偏离程度（SST）：
SST = Σ(yi - ȳ)²，其中yi表示第i个实际值，ȳ表示所有实际值的平均数。
该步骤计算的是所有实际值与平均值之间的总偏离程度。

2. 计算残差平方和（SSR）：
SSR = Σ(yi - ŷi)²，其中yi表示第i个实际值，ŷi表示第i个预测值。
该步骤计算的是所有实际值与预测值之间的总偏离程度。

3. 计算R-Squared指标：
R-Squared = 1 - SSR / SST
该步骤计算的是拟合线与实际数据点的总偏离程度的平方和与总变差程度的平方和的比值。

下面是Python中的实现示例：

import numpy as np

def r_squared(y_true, y_pred):
    # 计算SST
    sst = ((y_true - np.mean(y_true)) ** 2).sum()
    
    # 计算SSR
    ssr = ((y_true - y_pred) ** 2).sum()
    
    # 计算R-Squared
    r2 = 1 - (ssr / sst)
    
    return r2

三、R-Squared的应用场景

R-Squared指标在机器学习中的应用非常广泛，尤其是在回归问题中。通过计算R-Squared指标，我们可以评估模型的拟合程度，并可以按需对模型进行调整，以获得更准确的预测结果。

除此之外，R-Squared指标还可以用于比较不同模型的性能。通常情况下，R-Squared值越高的模型性能越好，我们可以据此选择最佳的回归模型。

四、总结

R-Squared是一项非常重要的评估指标，不仅在回归问题中有着广泛的应用，还可以帮助我们比较不同模型的性能。在Python中，计算R-Squared非常容易，只需要使用sklearn等相关库即可。此外，针对特定的问题，我们也可以编写自己的函数来计算R-Squared指标。