【机器学习实践一】搭建心脏病预测案例

首页 > 代码库 > 【机器学习实践一】搭建心脏病预测案例

【机器学习实践一】搭建心脏病预测案例

2024-08-27 09:33:21 220人阅读

产品地址：https://data.aliyun.com/product/learn?spm=a21gt.99266.416540.102.OwEfx2

一、背景

心脏病是人类健康的头号杀手。全世界1／3的人口死亡是因心脏病引起的，而我国，每年有几十万人死于心脏病。所以，如果可以通过提取人体相关的体侧指标，通过数据挖掘的方式来分析不同特征对于心脏病的影响，对于预测和预防心脏病将起到至关重要的作用。本文将会通过真实的数据，通过阿里云机器学习平台搭建心脏病预测案例。

二、数据集介绍

数据源： UCI开源数据集heart_disease
针对美国某区域的心脏病检查患者的体测数据，共303条数据。具体字段如下表：

字段名	含义	类型	描述
age	年龄	string	对象的年龄，数字表示
sex	性别	string	对象的性别，female和male
cp	胸部疼痛类型	string	痛感由重到无typical、atypical、non-anginal、asymptomatic
trestbps	血压	string	血压数值
chol	胆固醇	string	胆固醇数值
fbs	空腹血糖	string	血糖含量大于120mg/dl为true，否则为false
restecg	心电图结果	string	是否有T波，由轻到重为norm、hyp
thalach	最大心跳数	string	最大心跳数
exang	运动时是否心绞痛	string	是否有心绞痛，true为是，false为否
oldpeak	运动相对于休息的ST depression	string	st段压数值
slop	心电图ST segment的倾斜度	string	ST segment的slope，程度分为down、flat、up
ca	透视检查看到的血管数	string	透视检查看到的血管数
thal	缺陷种类	string	并发种类，由轻到重norm、fix、rev
status	是否患病	string	是否患病，buff是健康、sick是患病

三、数据探索流程

数据挖掘流程如下：
技术分享
整体实验流程：

1.数据预处理

数据预处理也叫作数据清洗，主要在数据进入算法流程前对数据进行去噪、填充缺失值、类型变换等操作。本次实验的输入数据包括14个特征和1个目标队列。需要解决的场景是根据用户的体检指标预测是否会患有心脏病，每个样本只有患病或不患病两种，是分类问题。因为本次分类实验选用的是线性模型逻辑回归，要求输入的特征都是double型的数据。
输入数据展示：
技术分享
我们看到有很多数据是文字描述的，在数据预处理的过程中我们需要根据每个字段的含义将字符型转为数值。
1）二值类的数据
二值类的比较容易转换，如sex字段有两种表现形式female和male，我们可以将female表示成0，把male表示成1。
2）多值类的数据
比如cp字段，表示胸部的疼痛感，我们可以通过疼痛的由轻到重映射成0~3的数值。

数据的预处理通过sql脚本来实现，具体请参考SQL脚本-1组件，


select age,
(case sex when ‘male‘ then 1 else 0 end) as sex,
(case cp when ‘angina‘ then 0  when ‘notang‘ then 1 else 2 end) as cp,
trestbps,
chol,
(case fbs when ‘true‘ then 1 else 0 end) as fbs,
(case restecg when ‘norm‘ then 0  when ‘abn‘ then 1 else 2 end) as restecg,
thalach,
(case exang when ‘true‘ then 1 else 0 end) as exang,
oldpeak,
(case slop when ‘up‘ then 0  when ‘flat‘ then 1 else 2 end) as slop,
ca,
(case thal when ‘norm‘ then 0  when ‘fix‘ then 1 else 2 end) as thal,
(case status  when ‘sick‘ then 1 else 0 end) as ifHealth
from  ${t1};