初级
OLS分解:SST=SSE+SSR验证
未完成
初级参考
完整示例代码供参考,建议自己理解后重新输入
import pandas as pd
import numpy as np
import statsmodels.api as sm
def solve():
url = 'https://liangdaima.com/static/data/wooldridge/wage1.csv'
df = pd.read_csv(url)
X = sm.add_constant(df['educ'])
y = df['wage']
model = sm.OLS(y, X).fit()
y_bar = y.mean()
SST = np.sum((y - y_bar)**2)
SSR = np.sum(model.resid**2)
SSE = SST - SSR
check = np.isclose(SST, SSE + SSR)
r2 = SSE / SST
print(check)
print(f'{r2:.4f}')
示例
输入
solve()
期望输出
True 0.1648
👑
升级 VIP
解锁全部题目,畅通无阻地学习
- ✓ 解锁全部训练包所有题目
- ✓ 查看完整参考代码和提示
- ✓ 浏览器内直接运行 Python 代码
- ✓ 自动批改 + 进度追踪
30天
¥18
1年
¥99
2年
¥158
3年
¥199