python计算方差膨胀因子_Python中的方差膨胀因子

最新推荐文章于 2024-08-23 07:00:00 发布

weixin_39527292

最新推荐文章于 2024-08-23 07:00:00 发布

阅读量818

点赞数

文章标签： python计算方差膨胀因子

本文介绍了如何在Python中计算方差膨胀因子(VIF)，用于评估多元线性回归中各解释变量之间的多重共线性。通过statsmodels库的variance_inflation_factor函数，结合add_constant方法或assign方法添加常量，计算并展示了各个变量的VIF值。

摘要由CSDN通过智能技术生成

正如其他人和Josef perktell在this post中所提到的，函数的作者variance_inflation_factor期望解释变量矩阵中存在常量。可以使用statsmodels中的add_constant将所需常量添加到数据帧，然后再将其值传递给函数。from statsmodels.stats.outliers_influence import variance_inflation_factor

from statsmodels.tools.tools import add_constant

df = pd.DataFrame(

{'a': [1, 1, 2, 3, 4],

'b': [2, 2, 3, 2, 1],

'c': [4, 6, 7, 8, 9],

'd': [4, 3, 4, 5, 4]}

)

X = add_constant(df)

>>> pd.Series([variance_inflation_factor(X.values, i)

for i in range(X.shape[1])],

index=X.columns)

const 136.875