您可以使用delta method 找到预测概率的近似方差。即,
var(proba) = np.dot(np.dot(gradient.T, cov), gradient)
其中gradient是模型系数预测概率的导数向量,cov是系数的协方差矩阵。
Delta 方法被证明对所有最大似然估计都渐近地起作用。但是,如果您的训练样本较小,渐近方法可能效果不佳,您应该考虑自举。
这是一个将 delta 方法应用于逻辑回归的玩具示例:
import numpy as np
import statsmodels.api as sm
import matplotlib.pyplot as plt
# generate data
np.random.seed(1)
x = np.arange(100)
y = (x * 0.5 + np.random.normal(size=100,scale=10)>30)
# estimate the model
X = sm.add_constant(x)
model = sm.Logit(y, X).fit()
proba = model.predict(X) # predicted probability
# estimate confidence interval for predicted probabilities
cov = model.cov_params()
gradient = (proba * (1 - proba) * X.T).T # matrix of gradients for each observation
std_errors = np.array([np.sqrt(np.dot(np.dot(g, cov), g)) for g in gradient])
c = 1.96 # multiplier for confidence interval
upper = np.maximum(0, np.minimum(1, proba + std_errors * c))
lower = np.maximum(0, np.minimum(1, proba - std_errors * c))
plt.plot(x, proba)
plt.plot(x, lower, color='g')
plt.plot(x, upper, color='g')
plt.show()
它绘制了以下漂亮的图片:
对于您的示例,代码将是
proba = logit.predict(age_range_poly)
cov = logit.cov_params()
gradient = (proba * (1 - proba) * age_range_poly.T).T
std_errors = np.array([np.sqrt(np.dot(np.dot(g, cov), g)) for g in gradient])
c = 1.96
upper = np.maximum(0, np.minimum(1, proba + std_errors * c))
lower = np.maximum(0, np.minimum(1, proba - std_errors * c))
plt.plot(age_range_poly[:, 1], proba)
plt.plot(age_range_poly[:, 1], lower, color='g')
plt.plot(age_range_poly[:, 1], upper, color='g')
plt.show()
它会给出以下图片
看起来很像一条蟒蛇,里面有一头大象。
您可以将其与 bootstrap 估计值进行比较:
preds = []
for i in range(1000):
boot_idx = np.random.choice(len(age), replace=True, size=len(age))
model = sm.Logit(wage['wage250'].iloc[boot_idx], age[boot_idx]).fit(disp=0)
preds.append(model.predict(age_range_poly))
p = np.array(preds)
plt.plot(age_range_poly[:, 1], np.percentile(p, 97.5, axis=0))
plt.plot(age_range_poly[:, 1], np.percentile(p, 2.5, axis=0))
plt.show()
delta 方法和 bootstrap 的结果看起来几乎一样。
然而,本书的作者走的是第三种方式。他们使用的事实是
概率 = np.exp(np.dot(x, params)) / (1 + np.exp(np.dot(x, params)))
并计算线性部分的置信区间,然后用logit函数进行变换
xb = np.dot(age_range_poly, logit.params)
std_errors = np.array([np.sqrt(np.dot(np.dot(g, cov), g)) for g in age_range_poly])
upper_xb = xb + c * std_errors
lower_xb = xb - c * std_errors
upper = np.exp(upper_xb) / (1 + np.exp(upper_xb))
lower = np.exp(lower_xb) / (1 + np.exp(lower_xb))
plt.plot(age_range_poly[:, 1], upper)
plt.plot(age_range_poly[:, 1], lower)
plt.show()
所以他们得到了发散区间:
这些方法产生如此不同的结果,因为它们假设不同的事物(预测概率和对数赔率)正态分布。即,delta 方法假设预测概率是正常的,而在书中,log-odds 是正常的。事实上,它们在有限样本中都不是正态的,在无限样本中它们都收敛到正态,但它们的方差同时收敛到零。最大似然估计对重新参数化不敏感,但它们的估计分布是,这就是问题所在。