由于我倾向于不使用 statsmodels 库,因此我创建了一个 Python 函数来执行 Breusch-Pagan 测试。它使用来自 SciKit-learn 的多元线性回归。
import numpy as np
from sklearn.linear_model import LinearRegression
from scipy.stats import chisqprob
def breusch_pagan_test(x, y):
'''
Breusch-Pagan test for heteroskedasticity in a linear regression model:
H_0 = No heteroskedasticity.
H_1 = Heteroskedasticity is present.
Inputs:
x = a numpy.ndarray containing the predictor variables. Shape = (nSamples, nPredictors).
y = a 1D numpy.ndarray containing the response variable. Shape = (nSamples, ).
Outputs a list containing three elements:
1. the Breusch-Pagan test statistic.
2. the p-value for the test.
3. the test result.
'''
if y.ndim != 1:
raise SystemExit('Error: y has more than 1 dimension.')
if x.shape[0] != y.shape[0]:
raise SystemExit('Error: the number of samples differs between x and y.')
else:
n_samples = y.shape[0]
# fit an OLS linear model to y using x:
lm = LinearRegression()
lm.fit(x, y)
# calculate the squared errors:
err = (y - lm.predict(x))**2
# fit an auxiliary regression to the squared errors:
# why?: to estimate the variance in err explained by x
lm.fit(x, err)
pred_err = lm.predict(x)
del lm
# calculate the coefficient of determination:
ss_tot = sum((err - np.mean(err))**2)
ss_res = sum((err - pred_err)**2)
r2 = 1 - (ss_res / ss_tot)
del err, pred_err, ss_res, ss_tot
# calculate the Lagrange multiplier:
LM = n_samples * r2
del r2
# calculate p-value. degrees of freedom = number of predictors.
# this is equivalent to (p - 1) parameter restrictions in Wikipedia entry.
pval = chisqprob(LM, x.shape[1])
if pval < 0.01:
test_result = 'Heteroskedasticity present at 99% CI.'
elif pval < 0.05:
test_result = 'Heteroskedasticity present at 95% CI.'
else:
test_result = 'No significant heteroskedasticity.'
return [LM, pval, test_result]