vif calculations

while(True):
    vif_df = pd.DataFrame()
    vif_df['VIF'] = [variance_inflation_factor(X.values, i) for i in range(X.shape[1])]
    vif_df['attributes'] = X.columns
    vif_df=vif_df.sort_values('VIF', ascending=False)
    if vif_df['VIF'].values[0]>6:
        print("Removing :",vif_df['attributes'].values[0],"as VIF value is", vif_df['VIF'].values[0])
        X=X.drop(columns=vif_df['attributes'].values[0]).reset_index(drop=True)
        continue
    else:
        break