导入类库
1 import numpy as np
2 import pandas as pd
3 from pandas import Series, DataFrame
4 import matplotlib.pyplot as plt
5 from sklearn.preprocessing import StandardScaler
6 from imblearn.over_sampling import SMOTE
7 from sklearn.ensemble import GradientBoostingClassifier
8 from sklearn.model_selection import train_test_split
9 from sklearn.linear_model import LogisticRegression
10 from sklearn.metrics import confusion_matrix
11 import itertools
12 from sklearn.model_selection import GridSearchCV
13 from sklearn.metrics import auc, roc_curve
作图函数
1 def plot_confusion_matrix(cm, classes,
2 title='Confusion matrix',
3 cmap=plt.cm.Blues):
4 """
5 This function prints and plots the confusion matrix.
6 """
7 plt.imshow(cm, interpolation='nearest', cmap=cmap)
8 plt.title(title)
9 plt.colorbar()
10 tick_marks = np.arange(len(classes))
11 plt.xticks(tick_marks, classes, rotation=0)
12 plt.yticks(tick_marks, classes)
13
14 threshold = cm.max() / 2.
15 for i, j in itertools.product(range(cm.shape[0]), range(cm.shape[1])):
16 plt.text(j, i, cm[i, j],
17 horizontalalignment="center",
18 color="white" if cm[i, j] > threshold else "black") # 若对应格子上面的数量不超过阈值则,上面的字体为白色,为了方便查看
19
20 plt.tight_layout()
21 plt.ylabel('True label')
22 plt.xlabel('Predicted label')
23 plt.show()
数据获取与解析
数据为结构化数据,不需要抽特征转化, 但特征Time和Amount的数据规格和其他特征不一样, 需要对其做特征做特征缩放
1 credit = pd.read_csv('./creditcard.csv')
2
3 print('原始行列 >>>>', credit.shape) # (284807行, 31列)
4 # print(credit.head()) # 前5行
5 # print(credit.dtypes) # 查看特征(列)类型。结果:数据类型只有float64和int64
6 # print(credit.isnull().any()) # 判断是否有缺失值。结果:无缺失值,方便后续处理
7 # print(credit.info()) # 查看数据集详细信息(类型,占用大小,缺失值,行列等)
特征工程
1 # c_counts = credit['Class'].value_counts()
2 # print(c_counts, type(c_counts)) # 对Class列分类统计,并判断类型
3 # print(c_counts.index, c_counts.values) # 提取索引和值
4 '''
5 结果:
6 0 284315
7 1 492
8 Name: Class, dtype: int64
9 Name: Class, dtype: int64 <class 'pandas.core.series.Series'>
10 Int64Index([0, 1], dtype='int64') [284315 492]
11 '''
12
13 # 对c_counts作图进行分析
14 # plt.figure(figsize=(10, 6))
15 # 饼图:两种作图方式
16 # ax = plt.subplot(121)
17 # c_counts是pandas的Series类型,pandas可以使用plot快速作图
18 # c_counts.plot(kind='pie', autopct='%0.3f%%', ax=ax)
19 # plt.pie(c_counts, autopct='%0.3f%%')
20
21 # 柱状图:两种作图方式
22 # ax = plt.subplot(122)
23 # c_counts.plot(kind='bar', ax=ax)
24 # plt.bar(c_counts.index, c_counts.values)
25 # plt.show()
26 '''
27 存在492例盗刷,占总样本的0.17%,
28 存在明显的数据类别不平衡问题,
29 可采用过采样(增加数据)的方法处理该问题
30 '''
特征转换
将时间从单位每秒化为单位每小时 divmod(7201,3600) 结果:(2, 1) 元组,2为商,1为余数
1 credit['Time'] = credit['Time'].map(lambda x: divmod(x, 3600)[0])
2 # print(credit['Time']) # map高级函数:将Time中的每个元素作用于lambda函数
特征选择
1 # Class列中值为0的为True,值为1为False,生成的cond0行数不变
2 # cond0 = credit['Class'] == 0
3 # Class列中值为0的为False,值为1为True,生成的cond0行数不变
4 # cond1 = credit['Class'] == 1
5 # print('cond0 >>>>', len(cond0))
6 # print('cond1 >>>>', len(cond1))
7
8 # 作图分析
9 # credit['V1'][cond0].plot(kind='hist', bins=500)
10 # credit['V1'][cond1].plot(kind='hist', bins=50)
11 # plt.show()
12
13 # 调试查看用
14 # print("credit['V1'] >>>>", credit['V1'])
15 # print('cond0 >>>>', cond0)
16 # print('cond1 >>>>', cond1)
17
18 # 筛选出存在于V1列中且在cond0中为True的值(284315)
19 # print("credit['V1'][cond0] >>>>", credit['V1'][cond0])
20 # 筛选出存在于V1列中且在cond0中为True的值(492)
21 # print("credit['V1'][cond1] >>>>", credit['V1'][cond1])
22
23 ''' 作图分析:将每一个特征根据Class的真假进行划分, 图像中两种图形的重合度越大说明该特征对Class的影响越小, 所以需要剔除掉无用的特征 '''
24 # cols = ['V1', 'V2', 'V3', 'V4', 'V5', 'V6', 'V7', 'V8', 'V9', 'V10',
25 # 'V11', 'V12', 'V13', 'V14', 'V15', 'V16', 'V17', 'V18', 'V19', 'V20',
26 # 'V21', 'V22', 'V23', 'V24', 'V25', 'V26', 'V27', 'V28']
27 # 作图:28行,1列,每一行显示一个特征对应的图
28 # plt.figure(figsize=(12, 2800))
29 # for i, col in enumerate(cols):
30 # ax = plt.subplot(28, 1, i + 1)
31 # density(normed)标准化数据:将过大或过小的数据统一标准化
32 # credit[col][cond0].plot(kind='hist', bins=500, density=True, ax=ax)
33 # credit[col][cond1].plot(kind='hist', bins=50, density=True, ax=ax)
34 #
35 # ax.set_title(col)
36 # plt.show()
37
38 # 待剔除的列(10列)
39 drops = ['V13', 'V15', 'V20', 'V22', 'V23', 'V24', 'V25', 'V26', 'V27', 'V28']
40 # 删除指定列(axis=1按列,axis=0按行)
41 credit2 = credit.drop(labels=drops, axis=1)
42 print('人眼剔除无用列后 >>>>', credit2.shape)
43 ''' 不同变量在信用卡被盗刷和信用卡正常的不同分布情况, 选择在不同信用卡状态下的分布有明显区别的变量。
因此剔除变量V13 、V15 、V20 、V22、 V23 、V24 、V25 、V26 、V27 和V28变量 '''
特征缩放
Amount变量和Time变量的取值范围与其他变量相差较大, 所以要对其进行特征缩放
1 # print('原Amount数据最大值', credit2['Amount'].max())
2 # print('原Amount数据最小值', credit2['Amount'].min())
3 # print('原Time数据最大值', credit2['Time'].max())
4 # print('原Time数据最小值', credit2['Time'].min())
5
6 # 创建标准化对象
7 standScaler = StandardScaler()
8 cols = ['Time', 'Amount']
9 # 标准化数据
10 credit2[cols] = standScaler.fit_transform(credit2[cols])
11 # print('标准化Amount后最大值 >>>>', credit2['Amount'].max())
12 # print('标准化Amount后最小值 >>>>', credit2['Amount'].min())
13 # print('标准化Time后最大值 >>>>', credit2['Time'].max())
14 # print('标准化Time后最小值 >>>>', credit2['Time'].min())
特征重要性排序
对特征的重要性进行排序,以进一步减少变量 利用GBDT梯度提升决策树进行特征重要性排序
1 # 创建GBDT对象
2 # clf = GradientBoostingClassifier()
3 # 特征训练集:前20列
4 # X_train = credit2.iloc[:, :-1]
5 # print('X_train.shape >>>>', X_train.shape)
6 # cols = X_train.columns
7 # print('X_train.columns >>>>', X_train.columns)
8 # 目标值训练集:Class列
9 # y_train = credit2['Class'] # y_train = credit2.iloc[:,-1]
10 # print('y_train.shape >>>>', y_train.shape)
11 # 训练数据
12 # clf.fit(X_train, y_train)
13 # 得到特征重要性数据
14 # feature_importances_ = clf.feature_importances_
15 # print('feature_importances_ >>>>', feature_importances_)
16 # 从大到小对特征重要性进行排序,并作图分析
17 # argsort():对数组排序并返回排序后每个元素对应的未排序时自身所在的索引
18 # index = feature_importances_.argsort()[::-1]
19 # print('从大到小排列特征重要性,返回每个元素的原索引 >>>>', index, len(index))
20
21 # plt.figure(figsize=(12, 9))
22 # 柱状图,第二个参数代表按从大到小排列的特征数据
23 # plt.bar(np.arange(len(index)), feature_importances_[index])
24 # 柱状图x坐标:第二个参数是按特征值从大到小排列后的特征名
25 # plt.xticks(np.arange(len(index)), cols[index])
26 # plt.show()
27 # 根据图像得到要删除的特征列(最小的后9列)
28 drops = ['V7', 'V21', 'V8', 'V5', 'V4', 'V11', 'V19', 'V1', 'Amount']
29 credit3 = credit2.drop(labels=drops, axis=1)
30 print('通过GBDT分析剔除无用列后 >>>>', credit3.shape)
31 # print('credit3.columns >>>>', credit3.columns)
模型训练
处理样本不平衡问题
目标变量“Class”正常和被盗刷两种类别的数量差别较大,会对模型学习造成困扰。
举例来说,假如有100个样本,其中只有1个是被盗刷样本,
其余99个全为正常样本,那么学习器只要制定一个简单的方法:
即判别所有样本均为正常样本,就能轻松达到99%的准确率。
而这个分类器的决策对我们的风险控制毫无意义。
因此,在将数据代入模型训练之前,我们必须先解决样本不平衡的问题。
现对该业务场景进行总结如下:
过采样(oversampling):
增加正样本使得正、负样本数目接近,然后再进行学习。
欠采样(undersampling):
去除一些负样本使得正、负样本数目接近,然后再进行学习。
本次处理样本不平衡采用的方法是过采样,
具体操作使用SMOTE(Synthetic Minority Oversampling Technique),
SMOET的基本原理是:
采样最邻近算法,计算出每个少数类样本的K个近邻,
从K个近邻中随机挑选N个样本进行随机线性插值,
构造新的少数样本,同时将新样本与原数据合成,产生新的训练集。
1 # SMOTE 过采样
2 X = credit3.iloc[:, :-1]
3 y = credit3.Class
4 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
5 X_train,y_train 作为训练数据 训练时,保证样本均衡,将X_train和y_train样本过采样处理 测试时候,可以样本不均衡
6 # print('未均衡的y训练集分类统计(Class) >>>>', y_train.value_counts())
7
8 smote = SMOTE()
9 # ndarray
10 X_train_new, y_train_new = smote.fit_sample(X_train, y_train)
11 # print('均衡后的x训练集 >>>>', X_train_new, type(X_train_new))
12 # print('均衡后的y训练集(Class) >>>>', y_train_new, type(y_train_new), len(y_train_new))
13 # y_train_new类型为numpy.ndarray,需转化为pandas.Series类型才可分类统计
14 # print('均衡后的y训练集分类统计(Class) >>>>', Series(y_train_new).value_counts())
求召回率
单独的逻辑回归求得查全率Recall rate,Recall也叫召回率
1 # 创建逻辑回归对象
2 # logistic = LogisticRegression()
3 # print(logistic)
4 '''
5 LogisticRegression(C=1.0, class_weight=None, dual=False, fit_intercept=True,
6 intercept_scaling=1, max_iter=100, multi_class='ovr', n_jobs=1,
7 penalty='l2', random_state=None, solver='liblinear', tol=0.0001,
8 verbose=0, warm_start=False)
9 '''
10 # 训练均衡后的数据
11 # logistic.fit(X_train_new, y_train_new)
12 # 预测
13 # y_ = logistic.predict(X_test)
14 # print('y_test >>>>', y_test)
15 # print('预测的y_ >>>>', y_)
16 # 交叉表
17 # print('交叉表 >>>>', pd.crosstab(y_test, y_, margins=True))
18
19 # 混合矩阵
20 # cm = confusion_matrix(y_test, y_)
21 # print('混合矩阵 >>>>', cm, type(cm))
22 # Recall------“正确被检索的正样本item(TP)"占所有"应该检索到的item(TP+FN)"的比例
23 # plot_confusion_matrix(cm, [0, 1], title='Recall:%0.3f' % (cm[1, 1] / (cm[1, 0] + cm[1, 1])))
交叉验证与调优
1 logistic = LogisticRegression()
2 clf = GridSearchCV(logistic, param_grid={'tol': [1e-3, 1e-4, 1e-5], 'C': [1, 0.1, 10, 100]}, cv=10, iid=False, n_jobs=1)
3 print(clf.fit(X_train_new, y_train_new))
4 # print('best_score_ >>>>', clf.best_score_)
5 # print('best_params_ >>>>', clf.best_params_)
6 # print('best_index_ >>>>', clf.best_index_)
7 # print('best_estimator_ >>>>', clf.best_estimator_)
8
9 # 预测
10 # y3_ = clf.best_estimator_.predict(X_test)
11 # print('y3_预测(best_estimator_) >>>>', confusion_matrix(y_test, y3_))
12
13 # y2_ = clf.predict(X_test)
14 # print('y2_预测 >>>>', confusion_matrix(y_test, y2_))
15
16 # cm2 = confusion_matrix(y_test, y2_)
17
18 # 可视化,对比逻辑斯蒂回归和GridSearchCV结果
19 # plot_confusion_matrix(cm, [0, 1], title='Logistic Recall:%0.3f' % (cm[1, 1] / (cm[1, 0] + cm[1, 1])))
20 # plot_confusion_matrix(cm2, [0, 1], title='GridSearchCV Recall:%0.3f' % (cm2[1, 1] / (cm2[1, 0] + cm2[1, 1])))
模型评估
解决不同的问题,通常需要不同的指标来度量模型的性能。
例如我们希望用算法来预测癌症是否是恶性的,
假设100个病人中有5个病人的癌症是恶性,
对于医生来说,尽可能提高模型的查全率(recall)比提高查准率(precision)更为重要,
因为站在病人的角度,发生漏发现癌症为恶性比发生误 判为癌症是恶性更为严重
由此可见就上面的两个算法而言,明显lgb过拟合了,
考虑到样本不均衡问题,
故应该选用简单一点的算法(逻辑回归)来减少陷入过拟合的陷阱
1 y_proba = clf.predict_proba(X_test)
2 # 预测被盗刷的概率
3 print(y_proba)