python 5-1逻辑回归代码报错

 

关注微信公共号：小程在线

关注CSDN博客：程志伟的博客

 

环境：python 3.6

#-*- coding: utf-8 -*-

#逻辑回归 自动建模
import pandas as pd

#参数初始化
filename = '../data/bankloan.xls'
data = pd.read_excel(filename)
x = data.iloc[:,:8].as_matrix()
y = data.iloc[:,8].as_matrix()


from sklearn.linear_model import LogisticRegression as LR
from sklearn.linear_model import RandomizedLogisticRegression as RLR 
rlr = RLR() #建立随机逻辑回归模型，筛选变量
rlr.fit(x, y) #训练模型
rlr.get_support() #获取特征筛选结果，也可以通过.scores_方法获取各个特征的分数
print(u'通过随机逻辑回归模型筛选特征结束。')

print(u'有效特征为：%s' % ','.join(data.columns[rlr.get_support()]))

错误：IndexError: boolean index did not match indexed array along dimension 0; dimension is 9 but corresponding boolean dimension is 8。

解决方法：

新建一个数据集，删除最后一列

data2=data.drop(u'违约',1)
print(u'有效特征为：%s'%','.join(data2.columns[rlr.get_support()]))
 
x=data[data2.columns[rlr.get_support()]].as_matrix()
lr=LR()
lr.fit(x,y)
print(u'逻辑回归模型训练结束')
print(u'模型的平均正确率：%s'%lr.score(x,y))