算法集锦(3)|采用医疗数据预测糖尿病的算法

2020-08-04 15:02:15 浏览数 (1)

糖尿病是一组以高血糖为特征的代谢性疾病。糖尿病时长期存在的高血糖,导致各种组织,特别是眼、肾、心脏、血管、神经的慢性损害、功能障碍。本文将介绍如何利用机器学习与医疗数据来预测个人患糖尿病的算法,在此过程中,我们还会学习如何进行数据准备、数据清洗、特征选择、模型选择盒模型计算。

计算条件:

  • Python 3.
  • Anaconda(Scikit Learn, Numpy, Pandas, Matplotlib, Seaborn)
  • Jupyter Notebook
  • 对监督机器学习的基本理解,特别是分类。

步骤1:数据集准备

准备数据集是一件很枯燥的事情,即使有大量的数据,有时也很难找到一个适用于待解决问题的数据集。本文中,我们直接使用UCI机器学习库中的“Pima Indians Diabetes Database”,我们将用机器学习算法来处理它。

步骤2:数据分析

导入数据后,首先我们应该对数据集进行分析,从而更好的理解数据和数据集的特征,以便确定是否进行数据清理。

首先,导入必要的计算库。

代码语言:javascript复制
%matplotlib inline
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
代码语言:javascript复制
diabetes = pd.read_csv('datasets/diabetes.csv')
diabetes.columns 

然后,利用pandas的head()方法来检查数据库。

代码语言:javascript复制
diabetes.head()

数据集的维度用以下方法获得。

代码语言:javascript复制
print("Diabetes data set dimensions : {}".format(diabetes.shape))

数据集的维度显示为(768,9)。“Outcome”列标识病人是否患有糖尿病,1代表患病,0代表健康。我们可以计算出768人中,500个标识为0(未患病),268个标识为1(患病)。

代码语言:javascript复制
diabetes.groupby('Outcome').size()

下面,我们利用pandas的数据可视化模块对数据集进行分析,查看数据的分布特征。

代码语言:javascript复制
diabetes.groupby(‘Outcome’).hist(figsize=(9, 9))

步骤3:数据清理

数据清理过程中,需要考虑一下几个方面。

(1)重复或无关的数据

(2)错误标识的数据,或者多次出现相同的标识

(3)缺失或空的数据点

(4)异常值

因为我们使用的是标准数据库,所以可以假定第(1)、(2)条已经被处理过了。所以我们重点考察缺失的数据点和异常值。

缺失或空数据点

可以采用如下的函数来搜寻缺失或空缺的数据点。

代码语言:javascript复制
diabetes.isnull().sum()
diabetes.isna().sum()

结果并未发现空缺数据点,如果存在的化,我们就需要对其进行相应的处理。

异常值

分析直方图时,我们发现某些列存在一些异常值,所以需要进行深入分析并确定如何处理它们。

血压(Blood pressure):通过分析数据,我们发现有些血压值为0。很明显,一个正常人的血压不可能为0,所以这些数据是错误的。

代码语言:javascript复制
print("Total : ", diabetes[diabetes.BloodPressure == 0].shape[0])
代码语言:javascript复制
Total :  35
代码语言:javascript复制
print(diabetes[diabetes.BloodPressure == 0].groupby('Outcome')['Age'].count())
代码语言:javascript复制
Outcome
0    19
1    16
Name: Age, dtype: int64

血糖水平(Plasma glucose levels): 同样,人的血糖水平也不可能降至0,所以数据集中有5列血糖值异常。

代码语言:javascript复制
print("Total : ", diabetes[diabetes.Glucose == 0].shape[0])
代码语言:javascript复制
Total :  5
代码语言:javascript复制
print(diabetes[diabetes.Glucose == 0].groupby('Outcome')['Age'].count())
代码语言:javascript复制
Total :  5
Outcome
0    3
1    2
Name: Age, dtype: int64

皮褶厚度(Skin Fold Thickness): 正常人的皮褶厚度一般不会小于10mm,可以发现该参数共出现227次0值。

代码语言:javascript复制
print("Total : ", diabetes[diabetes.SkinThickness == 0].shape[0])
代码语言:javascript复制
Total :  227
代码语言:javascript复制
print(diabetes[diabetes.SkinThickness == 0].groupby('Outcome')['Age'].count())
代码语言:javascript复制
Outcome
0    139
1     88
Name: Age, dtype: int64

身体质量指数(BMI): 除非一个人体重降低到威胁生命的地步,否则BMI值不为0或者接近0。

代码语言:javascript复制
print("Total : ", diabetes[diabetes.BMI == 0].shape[0])
代码语言:javascript复制
Total :  11
代码语言:javascript复制
print(diabetes[diabetes.BMI == 0].groupby('Outcome')['Age'].count())
代码语言:javascript复制
Outcome
0    9
1    2
Name: Age, dtype: int64

胰岛素(Insulin): 在极特殊的情况下,人体胰岛素值才会降低到0,而我们发现共有374项数据的胰岛素值出现0值,这显然是不正常的。

代码语言:javascript复制
print("Total : ", diabetes[diabetes.Insulin == 0].shape[0])
代码语言:javascript复制
Total :  374
代码语言:javascript复制
print(diabetes[diabetes.Insulin == 0].groupby('Outcome')['Age'].count())
代码语言:javascript复制
Outcome
0    236
1    138
Name: Age, dtype: int64

对于以上的异常数据,可以采用以下几种方法进行处理:

  1. 移除异常值:通常该方法难以实现,因为移除数据意味着会丢失有价值的数据。本例中,皮褶厚度和胰岛素两列出现了大量的异常值,若移除它们,则会丢失其他列的有效数据。
  2. 采用平均值: 该方法对于某些数据集是适用的,但对于本例来说,对血压项设置为平均值会给模型引入较大的误差。
  3. 弃用特征: 对于出现大量异常值的特征,有时可考虑弃用该特征(如皮褶厚度),但通过较难判断是否会影响模型的准确性。

通过分析数据,我们可以得知采用的数据集并不完整。经过综合分析,因为本例仅是为了验证算法的可行性,所以我们决定移除血压、BMI和血糖各特征中为0值的行。

代码语言:javascript复制
diabetes_mod = diabetes[(diabetes.BloodPressure != 0) & (diabetes.BMI != 0) & (diabetes.Glucose != 0)]
代码语言:javascript复制
print(diabetes_mod.shape)
代码语言:javascript复制
(724, 9)

步骤4:特征工程 特征过程是指将数据转换成特征,从而更好的去建模、优化和提高准确率的过程。 对于本文的算例,我们采用的是成熟的数据集,无法进一步的创建或消除任何数据点了,所以我们选择以下特征创建模型。

‘Pregnancies’, ‘Glucose’, ‘Blood Pressure’, ‘Skin Thickness’, ‘Insulin’, ‘BMI’, ‘Diabetes Pedigree Function’, ‘Age’

粗略的看来,皮褶厚度对于预测糖尿病可能不是一个预测参数,但其未必对模型的预测结果没有作用。所以我们采用了数据集全部的特征,并将其设置为“X变量”,而将预测结果设置为“Y变量”。

代码语言:javascript复制
feature_names = ['Pregnancies', 'Glucose', 'BloodPressure', 'SkinThickness', 'Insulin', 'BMI', 'DiabetesPedigreeFunction', 'Age']
代码语言:javascript复制
X = diabetes_mod[feature_names]
y = diabetes_mod.Outcome

通常,特征工程会在选择模型前实施。但本文中,我们采取一个不同的策略,我们先将数据集中所有的特征放入模型中,后续再详细的讨论各个特征对于模型的重要性。

步骤5:模型选择

模型选择或算法选择是机器学习中最有趣和最核心的部分。在该环节,我们会选择出对数据集表现最好的模型(算法)来进行预测。

我们会计算不同分类模型(在默认参数下)的分类准确率(或测试准确率),从而确定对数据集拟合最优的模型。首先,我们导入7种不同的分类器,分别为:K-Nearest Neighbors, Support Vector Classifier, Logistic Regression, Gaussian Naive Bayes, Random Forest and Gradient Boost。

代码语言:javascript复制
from sklearn.neighbors import KNeighborsClassifier
from sklearn.svm import SVC
from sklearn.linear_model import LogisticRegression
from sklearn.tree import DecisionTreeClassifier
from sklearn.naive_bayes import GaussianNB
from sklearn.ensemble import RandomForestClassifier
from sklearn.ensemble import GradientBoostingClassifier

然后,将各个分类器按默认参数初始化,并建立一个模型列表。

代码语言:javascript复制
models = []
代码语言:javascript复制
models.append(('KNN', KNeighborsClassifier()))
models.append(('SVC', SVC()))
models.append(('LR', LogisticRegression()))
models.append(('DT', DecisionTreeClassifier()))
models.append(('GNB', GaussianNB()))
models.append(('RF', RandomForestClassifier()))
models.append(('GB', GradientBoostingClassifier()))

计算方法

为了避免过拟合的发生,我们通常开展以下两方面的工作。

  1. 训练/测试数据划分
  2. K折叠交叉验证(K-Fold Cross Validation)

这里,我们将用“train_test_split”函数进行数据划分,“cross_val_score”函数进行K折叠交叉验证。通过这两个步骤,我们可以确定适用于数据集的最优分类器模型。

代码语言:javascript复制
from sklearn.model_selection import train_test_split
from sklearn.model_selection import cross_val_score
from sklearn.metrics import accuracy_score

训练/测试数据划分

通过划分,我们将数据集分为两个部分,训练数据集(Training set)和测试数据集(Testing set)。训练数据集用来训练模型,测试数据集用来评估模型的准确率。

代码如下:

代码语言:javascript复制
X_train, X_test, y_train, y_test = train_test_split(X, y, stratify = diabetes_mod.Outcome, random_state=0)

然后我们用“accuracy_score”来计算各个模型的准确率。

代码语言:javascript复制
names = []
scores = []
代码语言:javascript复制
for name, model in models:
    model.fit(X_train, y_train)
    y_pred = model.predict(X_test)
    scores.append(accuracy_score(y_test, y_pred))
    names.append(name)
代码语言:javascript复制
tr_split = pd.DataFrame({'Name': names, 'Score': scores})
print(tr_split)

K折叠交叉验证

对于个分类或回归问题,假设有多个可选的模型为。K-折叠交叉验证就是将训练集的1/k作为测试集,每个模型训练k次,测试k次,错误率为k次的平均,最终选择平均率最小的模型Mi。

该过程可以利用 Sickit_Learn库的如下代码实现:

代码语言:javascript复制
names = []
scores = []
代码语言:javascript复制
for name, model in models:
    
    kfold = KFold(n_splits=10, random_state=10) 
    score = cross_val_score(model, X, y, cv=kfold, scoring='accuracy').mean()
    
    names.append(name)
    scores.append(score)
代码语言:javascript复制
kf_cross_val = pd.DataFrame({'Name': names, 'Score': scores})
print(kf_cross_val)

各模型的准确率如下图所示。

代码语言:javascript复制
axis = sns.barplot(x = 'Name', y = 'Score', data = kf_cross_val)
axis.set(xlabel='Classifier', ylabel='Accuracy')
代码语言:javascript复制
for p in axis.patches:
    height = p.get_height()
    axis.text(p.get_x()   p.get_width()/2, height   0.005, '{:1.4f}'.format(height), ha="center") 
    
plt.show()

可以看到,二元回归算法(Logistic Regression)的准确率最高(77.64%),因此我们选择二元回归作为数据集的模型。

步骤6:模型参数调整

采用默认参数,二元回归模型获得了较好的预测准确率。接下来,我们将对模型的参数进行调整,优化模型,从而获得更准确的模型。本例采用的是GridSearchCV方法,该方法通过交叉验证对参数空间进行求解,寻找最佳的参数。

首先,导入GridSearchCV方法。

代码语言:javascript复制
from sklearn.model_selection import GridSearchCV

然后,给出二元回归模型的参数列表。

代码语言:javascript复制
# Specify parameters
c_values = list(np.arange(1, 10))
代码语言:javascript复制
param_grid = [
    {'C': c_values, 'penalty': ['l1'], 'solver' : ['liblinear'], 'multi_class' : ['ovr']},
代码语言:javascript复制
    {'C': c_values, 'penalty': ['l2'], 'solver' : ['liblinear', 'newton-cg', 'lbfgs'], 'multi_class' : ['ovr']}
]

将数据输入GridSearchCV,通过交叉验证来确认不同参数的组合效果。

代码语言:javascript复制
grid = GridSearchCV(LogisticRegression(), param_grid, cv=strat_k_fold, scoring='accuracy')
代码语言:javascript复制
grid.fit(X_new, y)

经过一系列的训练和评估,GridSearchCV给出了一些有用的信息用来寻找最优参数。

代码语言:javascript复制
print(grid.best_params_)
print(grid.best_estimator_)

经过分析,我们确定最优的超参数如下:

代码语言:javascript复制
{'C': 1, 'multi_class': 'ovr', 'penalty': 'l2', 'solver': 'liblinear'}

将优化后的参数输入回二元回归模型,可以看到模型预测的准确度得到了提升。

代码语言:javascript复制
logreg_new = LogisticRegression(C=1, multi_class='ovr', penalty='l2', solver='liblinear')
代码语言:javascript复制
initial_score = cross_val_score(logreg_new, X_new, y, cv=strat_k_fold, scoring='accuracy').mean()
print("Final accuracy : {} ".format(initial_score))
代码语言:javascript复制
Final accuracy : 0.7805877119643279

值得指出的是,本算例采用的是传统的机器学习算法,所以获得的预测准确率(78.05%)不尽理想。若采用深度神经网络模型,预测准确率应有较大的提升,后续文章会对此进行报道。

0 人点赞