python有没有计算方差膨胀因子的函数

请教大神，谁知道python如何计算方差膨胀因子，有没有现成的库(函数)调用,自己写代码实现也可以的，谢谢各位大神~~~

http://www.statsmodels.org/stabl ... ce_inflation_factor

1. from statsmodels.stats.outliers_influence import variance_inflation_factor
   
2. import numpy as np
   
3. 
   
4. data = [[1,2,3,4,5],
   
5.         [2,4,6,8,10],
   
6.         [1,1,1,1,1]]
   
7. X = np.array(data).T
   
8. 
   
9. variance_inflation_factor(X,2)

复制代码

fengxingliulizi 发表于 2017-9-10 20:41
http://www.statsmodels.org/stable/generated/statsmodels.stats.outliers_influence.variance_inflation_ ...

请问那个2是什么意思？

1. def cal_vif(df,col):
   
2.     df = sm.add_constant(df)
   
3.     col.append('const')
   
4.     df_vif = df.loc[:,col]
   
5.     return pd.DataFrame({'variable':col,'VIF':[variance_inflation_factor(np.matrix(df_vif),i) for i in range(len(col))]}).iloc[:-1]

复制代码

jungsee 发表于 2021-5-29 11:12

使用例子：
cal_vif(df_1,['INCOME','AGE', 'MALE'])