Statsmodels回归模型单值预测：处理常数项的正确姿势

在使用statsmodels库进行线性回归分析时，例如通过statsmodels.api.ols类，我们经常需要为自变量添加一个常数项（截距项），以便模型能够拟合出非零截距的线性关系。这通常通过statsmodels.api.add_constant函数来实现。然而，当模型训练完成后，许多用户在尝试对单个新数据点进行预测时，可能会遇到困惑或错误，尤其是不清楚如何正确处理这个常数项。

Statsmodels回归模型与常数项

statsmodels.api.OLS（Ordinary Least Squares，普通最小二乘法）是Statsmodels库中用于拟合线性回归模型的核心工具。在构建模型时，我们通常会准备一个因变量Y和一个自变量矩阵X。如果希望模型包含一个截距项，就需要手动将一个常数（通常是1）列添加到自变量矩阵X中。sm.add_constant()函数正是为此目的设计的。它会在输入数据的最前面（默认行为）添加一列值为1的常数。

例如，如果我们有一个原始特征向量x_raw，为了训练一个包含截距的线性模型，我们会这样准备自变量：

import statsmodels.api as sm
import numpy as np
import pandas as pd

# 1. 模拟数据
np.random.seed(42)
# 假设我们有一个特征 'Feature_X' 来预测 'Target_Y'
feature_data = np.random.rand(100) * 10 + 5 # 原始特征数据
target_data = 2 * feature_data + np.random.randn(100) * 2 + 10 # 目标数据

# 转换为DataFrame，模拟实际使用场景
df = pd.DataFrame({'Feature_X': feature_data, 'Target_Y': target_data})

# 2. 准备训练数据
Y_train = df['Target_Y']
X_raw_train = df['Feature_X']

# 为自变量添加常数项
X_train = sm.add_constant(X_raw_train)

# 3. 拟合OLS模型
model = sm.OLS(Y_train, X_train)
results = model.fit()

print("模型拟合结果摘要：")
print(results.summary())

在上述代码中，X_train的结构将是两列：第一列是常数1，第二列是Feature_X的实际值。

进行单值预测的挑战

当我们需要对一个全新的、单个特征值进行预测时，直观上可能会尝试直接将这个值传递给results.predict()方法。然而，这通常会导致错误或非预期的结果。results.predict()方法期望的输入（exog参数）是一个与训练时X矩阵具有相同列数和结构的数组或DataFrame。这意味着，如果训练时X包含了常数项，那么用于预测的输入数据exog也必须包含一个常数项。

Lumen5

一个在线视频创建平台，AI将博客文章转换成视频

105

查看详情

例如，如果我们想预测当Feature_X为7.5时的Target_Y值，直接传递7.5或np.array([7.5])是行不通的，因为它们缺少常数项，并且维度可能不匹配模型期望的二维结构。

正确的单值预测方法

解决这个问题的关键是确保用于预测的单个输入值也被正确地转换为与训练数据X相同的结构，即为其添加常数项，并保持其二维数组的形态。

以下是进行单值预测的正确步骤和示例代码：

1. 准备原始预测值： 确定您要预测的单个原始特征值。
2. 转换为二维数组： 将这个单个值放入一个列表中，然后（可选地）转换为NumPy数组，以确保其具有二维结构（例如 [7.5] 变为 [[7.5]]）。
3. 添加常数项： 使用sm.add_constant()函数为这个二维数组添加常数项。此时，sm.add_constant会将[1.0]作为第一列，原始值作为第二列，形成一个1xN（N为特征数+1）的矩阵。
4. 调用predict()方法： 将处理后的预测输入数据传递给results.predict()方法。

# 4. 进行单值预测
print("\n进行单值预测：")
# 假设我们要预测当原始特征值为 7.5 时的目标值
single_raw_feature_value = 7.5

# 正确的方式：为预测值添加常数项，并确保其维度与训练时的X匹配
# 注意：sm.add_constant 默认会添加一个常数项列
# 对于单个预测值，我们需要将其放入一个列表中，再转换为数组，然后添加常数项
# 这里的 `has_constant='add'` 参数确保即使输入数据看起来已经包含常数，也会强制添加。
# 对于一个全新的、不含常数项的单值，sm.add_constant([value]) 已经足够，但使用 'add' 更具普适性。
X_predict_single = sm.add_constant([single_raw_feature_value], has_constant='add')

predicted_value = results.predict(X_predict_single)
print(f"当原始特征值为 {single_raw_feature_value} 时，预测的目标值为：{predicted_value[0]:.4f}")

# 也可以预测多个值，原理相同
print("\n预测多个值：")
multiple_raw_feature_values = np.array([6.0, 8.5, 10.0])
# 对于多个值，sm.add_constant 会为每个值添加常数项
X_predict_multiple = sm.add_constant(multiple_raw_feature_values)
predicted_multiple_values = results.predict(X_predict_multiple)
print(f"当原始特征值为 {multiple_raw_feature_values} 时，预测的目标值为：{predicted_multiple_values}")

注意事项

• 维度匹配： results.predict()的exog参数必须是一个二维数组（或类似结构，如DataFrame），即使您只预测一个数据点。这是因为Statsmodels内部处理数据时，期望输入是样本数 x 特征数的矩阵。将单个值用列表包裹（[value]）即可满足这一要求。
• 常数项的一致性： 如果您的模型在训练时使用了sm.add_constant()添加常数项，那么在进行预测时，也必须对预测输入数据执行相同的操作。忽略这一点将导致预测结果不准确或程序报错。
• has_constant='add'参数： 在sm.add_constant()函数中，has_constant='add'参数指示函数总是添加常数项，即使它认为输入数据中可能已经存在常数项。对于从原始数据生成预测输入的情况，即使不明确指定'add'，默认行为通常也能正确添加常数项。但在某些复杂场景下，使用'add'可以增加代码的健壮性。
• 输入数据类型： 确保传递给predict()的exog参数

以上就是Statsmodels回归模型单值预测：处理常数项的正确姿势的详细内容，更多请关注php中文网其它相关文章！