泰坦尼克船员生存预测

简介: 泰坦尼克船员生存预测

1.任务背景


1.赛题任务


泰坦尼克号的沉没是历史上最为声名狼藉的沉船事件之一。1912 年 4 月 15 日,被广泛认为是“永不沉没的”泰坦尼克号在处女航中与冰山相撞后沉没。不幸的是,船上没有足够的救生艇,导致 2224 名乘


客和船员中有 1502 人死亡。虽然生存需要一些运气因素,但似乎有些人群比其他人更有可能生存下来。在这个挑战中,我们要求您建立一个预测模型来回答这个问题:“什么类型的人更有可能生存?”使用乘客数据(如姓名、年龄、性别、社会经济阶层等)。


2.赛题数据


在此任务中数据被分为两组:


(1)训练数据将提供:


Survival: 人员是否存活(0,1)


Pclass: 船票等级(1,2,3)


Sex: 性别(male,female)


Age: 年龄


Sibsp: 兄弟姐妹/配偶数


Parch: 父母孩子数


Ticket: 门票号码


Fare: 乘客收费


Cabin: 仓位号码


Embarked: 启程港(C,Q,S)


(2)测试数据将提供:


Pclass: 船票等级(1,2,3)


Sex: 性别(male,female)


Age: 年龄


Sibsp: 兄弟姐妹/配偶数


Parch: 父母孩子数


Ticket: 门票号码


Fare: 乘客收费


Cabin: 仓位号码


Embarked: 启程港(C,Q,S)


2.数据分析与处理


1.数据预处理

我们先将训练数据集与测试数据集通过 python 的 pandas 进行了数据的导入并且将训练集与测试集进行了合并。之后对各个数据字段进行了缺失值的统计。结果如下:

a9f4e9bccaf6b170707d1531a59f5074.png

图 1


我们发现对于船舱位置(cabin)和年龄(age)特性的缺失值比较多。为了进一步研究各个特征对于乘客是否生存的影响,此处先进行了简单的平均值填充缺失值的方法将各个字段进行了缺失值处理。当然,这并不是我们最终对缺失值的处理,此处进行简单的缺失值填充是为了方便我们进一步对数据进行分析。处理后缺失情况如下:

fd9d6e9b2dc2ec68451aebd038a75882.png

图 2


2.数据分析


在对数据进行了简单的预处理以后,现在的数据已经比较完整了。这方便了我们对数据做一些简单的可视化显示。所以,我们将各个特征值与生存的关系进行了简单的数据可视化。如下:


2d6522e1062e47bbd33072cbb47c5312.jpg


图 3


先总体来看的话,我们明显发现船票(ticket)和船舱位置(cabin)两个字段的特征并不明显,导致其与乘客是否存活的关联性并不能够明显看出。之后,我们再来仔细地分析每一个特征对乘客是否生存的影响程度。


对于船票等级(pclass),我们对该特征进行了进一步的可视化。如图 4 所示:

810a71e50d5f274260e91fbd45fd4347.jpg

图 4


我们先分别对 pclass 为 1,2,3 的等级的中乘客的存活人数做了统计并可视化显示,另外还对该特征与存活率之间的关系做了具体显示。从图中我们不难发现对于船舱等级为 1 的乘客,我们发现其中存活的乘客甚至比死亡的乘客要多。在通过观察船舱等级为 1 的乘客的存活率,发现其存活率高达 60% 以上。所以我们猜测船舱等级数越小乘客的船舱应该是越好,越高级,导致其存活率会比较高。为了验证我们的猜想,我们继续分析了船舱等级为 2 和 3 的乘客的存活情况,先做内部比较发现,等级为 2 的乘客的存活于死亡大致持平,死亡略高于存活人数,存活率大概也在 48% 左右。而再观察等级为 3 的乘客,发现死亡人数是存活人数的三倍多,而存活率也是低到了 28% 左右。所以,我们不难总结出,船舱等级越高的乘客(数越小),越容易存活,而船舱等级比较低的乘客不易存活。这可以证明 pclass 特征对乘客是否生存的影响是巨大的,所以这是我们不可忽视的特征。


接着,我们又对性别(sex)特征进行了分析。同 pclass 特征的处理方式,这里我们也是对数据进行了具体的可视化显示(如图 5):


95e5fb37180cadd0f0e06da4df8f99d2.jpg

图 5


为了避免男性和女性数据量的差异,所以我们对数据中的男女数量做了一个简单的统计。我们发现男性人数大概是女性人数的两倍,男性人数在 600 左右,而女性人数在 300 左右。再观察男性乘客与女性乘客的存活率,我们发现女性乘客虽然人数少,但是女性的存活率是男性的三倍多,所以我们可以确定性别对乘客存活情况的影响很大。这也是我们需要重视的数据,会对我们的预测结果产生很大的影响。


之后,我们又对船票价格(fare)进行了具体的分析。并做了进一步的可视化分析。如图 6 所示:


图 6


我们发现乘客的船票的价格分布大概在 30 左右,说明大部分人的船票价格偏低,而只有少一部分人的船票价格比较高。为了方便做数据的分析,我们又将船票价格从最低到最高进行了 4 等分,做了简单的分类,分成 low,mid,high,veryhigh 四类来代表船票价格等级。我们可以从图中发现,船票高的乘客的存活率会高,而且船票价格由低到高,乘客的存活率也是逐渐增长的。这证明了船票价格对乘客是否存活的影响程度很大,所以这一特征会对我们有所帮助。


我们对年龄(age)特征也进行了类似的相关分析与数据可视化。但考虑到年龄的数据缺失值比较多,而我们又是采取的平均值填充的方法去做的预处理,所以参考的价值相对前几个特性应该不是那么的大。具体数据可视化如图 7 所示:

2467e369dfec55a23d73a7f7ba1ed88f.png

图 7


左边的图描述了乘客年龄的分布,右边的图描述了年龄与存活率的关系。我们发现乘客中的中年人比较多,而存活情况是年幼的孩子存活率比较高,超过了死亡率,而 30 岁左右的中年人存活率很低,60 岁以上的老人存活率也不是很高。该特征值也会在之后的预测中产生一定的影响。然后,我们又对港口号(Embarked)进行了数据的分析和可视化的展示,如图 8:

f7181c9cc145fcc57b5b46e46591e0c6.jpg

图 8


我们发现 s 港口的人数比较多,C 和 Q 港口的人数比较少。再通过观察各个港口的乘客的存活情况,我们发现,各个港口的存活情况差距也巨大。从 s 港等船的人最多,但是存活率却是最低的,而从 Q 港登船的人虽然最少,但是存活率也不是很高,明显低于 50%。反而是登船人数居中的 C 港的存活情况比较乐观,存活人数大于死亡人数,存活率在 50% 以上。这使得我们对该特征与乘客存活的联系产生了很大的疑惑。因为只从港口号和乘客生存率来看,我们并不能得出什么结论。考虑到实际情况中,登船港口与地理位置有关系,而地理位置往往决定着一个人的职位与阶级,所以我们猜测港口号可能受到了其他特征的影响,然后我们又猜测应该是船舱等级(pclass)特征对港口号产生了一定程度的影响。于是,我们又分析了从各个港口登船的人的船舱等级分布。如图 9 所示:

d249a20088be4ee615945d083c7d8b5f.jpg

我们发现这些名字中所带的描述特征的信息与年龄有着一定的联系(图 12),并且可以提取与职业、阶级有关的信息作为幸存预测的特征。

8a0854888b758c5e2bb29f76545a225e.jpg


图 12


果然像我们所猜测的那样,从图中可以看出,从 s 港,Q 港登船的人中,船舱等级为 3 的人占比高,而 C 港登船的人中,船舱等级为 1 的人占比较高。所以此特征也可以作为辅助我们进行乘客是否存活的预测。


最后,我们也发现了在名字(name)字段中隐藏着一些重要的信息,这些信息往往代表着乘客的一些相关特征。这会再之后的数据处理中做详细的描述。


3.数据修改

现在,我们准备挖掘出名字(name)中的有用信息。首先,我们将 name 字段用逗号和空格分隔开,选择第一个项信息:

706588944b5e78b7b4ebaac4d1ac0f9b.png


并显示出提取出的 title 数量与年龄关系(图 10、11):

ab83a9300aca1364983cf1e0db2eb457.jpg


图 10

041d925fc20765241fa591b5a6937170.jpg


图 11

从图中我们可以看出名字中带有“Mr”的乘客幸存率较低,而“Mrs”、“Miss”等存活率较高,还有一些少数职业如“Capt”的存活率很高。我们将这 18 个特征化为 4 个更为有特点的新特征(图 13)。


d87b0395337d279e19914373b3a05581.jpg


图 13

对于 Cabin(船舱号)这个特征,缺失了太多的数据,但是对有限的数据进行分析,我们发现乘客的幸存与否与此特征有一定联系,并且在现实中船舱号也是判断生存率的一项重要特征,所以我们将缺失数据填为“U”,其他已知数据改为其首字母(图 14)。



93218f1abfae1e0eb1aa77f6dd5fcd5b.jpg


图 14

3.模型构建


1.随机森林

(1)原理介绍


要对一个输入的样本进行分类,将这个样本交给每棵树进行分类,根据所有树分类的结果选择分成哪一类。而每棵树的产生规则是在大小为 N 的训练集中有放回地抽取 N 个训练样本,作为该树的训练集,有放回和随机抽样保证了差异性与准确性。在得到了训练集后选取特征数开方个特征,在每次树进行分裂时从这些特征中选择最优的,可用算法有 ID3、C4.5、C5.0 等,每棵树都尽可能地生长不会进行剪枝。


随机选取训练集和随机选取特征成为了随机森林分类性能良好的主要条件,使得随机森林不易陷入过拟合,并且有很好的抗噪能力。


(2)代码实现


调用 sklearn 包中的随机森林分类器实现分类,具体代码如下:

1ec367c2bfac8e0e872c527b593c4fc8.jpg


其中参数部分(PRF)是已经通过 GridSearchCV 选出的较好参数,并没有将开始的备选参数展示出来。模型测试部分采用 10 折交叉验证取平均的方式,对训练集进行评估,在采用下图中特征为分类标准时结果如下:


5582be36f090d4ff39354437f0a407be.jpg

可见对于本地训练集可以达到 84% 的正确率,若采用以 F1 分数作为评分标准则可达到 0.77 分。


在最后的提交中,我们发现对于测试集的得分仅为其中对于参数的调节问题是通过 GridSearchCV 选出来的比较好的参数。模型的猜测部分同样采取了交叉验证的方式对训练集进行了评估。本地测试结果如下:


f3800de6b58c0f7c3d316de33d76341f.png

2.SVM


(1)简介


支持向量机(supportvectormachines,SVM)是一种二分类模型,它的基本模型是定义在特征空间上的间隔最大的线性分类器,间隔最大使它有别于感知机;SVM 还包括核技巧,这使它成为实质上的非线性分类器。SVM 的学习策略就是间隔最大化,可形式化为一个求解凸二次规划的问题,也等价于正则化的合页损失函数的最小化问题。SVM 的学习算法就是求解凸二次规划的最优化算法。一般 svm 有三种:硬间隔支持向量机(线性可分支持向量机):当训练数据线性可分时,可通过硬间隔最大化学得一个线性可分支持向量机。软间隔支持向量机:当训练数据近似线性可分时,可通过软间隔最大化学得一个线性支持向量机。非线性支持向量机:当训练数据线性不可分时,可通过核方法以及软间隔最大化学得一个非线性支持向量机。在本文中,对于乘客是否生存的预测,是一个标准的二分类问题,所以考虑选择 Svm 分类器实现。


(2)原理介绍)原理介绍)原理介绍 超平面:定义为 +=0,是能够将高维空间,是能够将高维空间一分为二的平面。要对一个输入的样本进行二分类,就相当于依据样本建立一个高维空间,然后利用 Karush-Kuhn-Tucker(KKT)条件和拉格朗日公式来推出边际最大化的划分超平面。对于线性不可分的数据可以利用内积核函数的方法来实现 svm 分类。同时,SVM 的最终决策函数只由少数的支持向量所确定,计算的复杂性取决于支持向量的数目,而不是样本空间的维数,这在某种意义上避免了“维数灾难”。(代码实现)


调用 sklearn 包中的 svm 分类器实现分类,具体代码如下:

1f88599e5921e51e3bf6b10beb968778.png


其中对于参数的调节问题是通过 GridSearchCV 选出来的比较好的参数。模型的猜测部分同样采取了交叉验证的方式对训练集进行了评估。本地测试结果如下:

210b8876994a343b4d3b89df3a260fd9.jpg


可见对于本地训练集可以达到 82% 的正确率,在最后的提交中,我们对于测试集的得分为 0.78468 分,通过 svm 模型和对数据的处理,我们成功预测了 78.4% 的乘客的生存情况。


3.CatBoost


1)原理介绍


我们可以使用 CatBoost,而不需要任何显式的预处理来将类别转换为数字。CatBoost 使用在各种统计上的分类特征和数值特征的组合将分类值转换成数字。它减少了对广泛的超参数调优的需求,并降低了过度拟合的几率,这也导致了模型变得更加具有通用性。但它还包含一些参数,比如树的数量、学习速率、正则化、树的深度等等。CatBoost 算法的设计初衷是为了更好的处理 GBDT 特征中的 categoricalfeatures。在处理 GBDT 特征中的 categoricalfeatures 的时候,最简单的方法是用 categoricalfeature 对应的标签的平均值来替换。在决策树中,标签平均值将作为节点分裂的标准。这种方法被称为 GreedyTarget-basedStatistics,简称 GreedyTS,这种方法有一个显而易见的缺陷,就是通常特征比标签包含更多的信息,如果强行用标签的平均值来表示特征的话,当训练数据集和测试数据集数据结构和分布不一样的时候会出条件偏移问题。一个标准的改进 GreedyTS 的方式是添加先验分布项,这样可以减少噪声和低频率类别型数据对于数据分布的影响,添加先验项是一个普遍做法,针对类别数较少的特征,它可以减少噪声数据。对于回归问题,一般情况下,先验项可取数据集 label 的均值。对于二分类,先验项是正例的先验概率。利用多个数据集排列也是有效的,但是,如果直接计算可能导致过拟合。CatBoost 利用了一个比较新颖的计算叶子节点值的方法,这种方式(oblivioustrees,对称树)可以避免多个数据集排列中直接计算会出现过拟合的问题。


4.GridSearchCV 调参器


GridSearchCV 根据其名字可以拆分为两部分,GridSearch 和 CV,即网格搜索和交叉验证。这两个名字都非常好理解。网格搜索,搜索的是参数,即在指定的参数范围内,按步长依次调整参数,利用调整的参数训练学习器,从所有的参数中找到在验证集上精度最高的参数,这其实是一个训练和比较的过程。而 CV 则是交叉验证,交叉验证原理是将训练数据随机等分为 n 份,选取一份作为本次的测试集,剩下的数据作为本次的训练集预测测试集的结果,重复选出测试集和训练集,直到全部数据被成功预测一遍,得到了对训练集的预测结果。并重新将数据进行等分,重复上边的过程,再得到一份训练集的预测结果,重新等分操作进行 n 次,最后训练集的预测结果为这 n 次预测的平均值,与训练集真实结果进行对比,得到训练集的预测效果。


我们采用 GridSearchCV 的目的是减少调参所花的时间,将更多的精力放在对数据的分析、处理与选择上,以达到更好的预测效果。


4.总结与分工


在本次任务中,随机森林的表现最好,成功正确预测了测试集 79.9% 的乘客的生成情况。但是我们只是单独地使用了三个模型,并没有将它们整合到一起,若是使用软投票或是硬投票的方式综合考虑三个模型预测的结果,我们相信预测效果会更好。通过本次课程,我们学到了很多有关人工智能的知识,而这次的任务更是让我们有了实践的机会,参加 kaggle 这种外国知名的比赛,开阔了我们的视野,一次次地提交与修改代码令我们更加深刻的意识到自己的不足,这也成为了我们不断前进的动力。


在分工方面,对于报告的撰写由我们三位同学共同完成,基本占比一样。在代码部分我们一个人负责一个模型,孙浩然负责随机森林,王润欣负责 Svm,李子易负责 Catboost,数据的读入和基本处理由孙浩然完成,在对数据进行进一步处理与分析时由三个人共同完成。


相关文章
|
6月前
|
机器学习/深度学习 算法 安全
泰坦尼克号乘客生存情况预测分析2
泰坦尼克号乘客生存情况预测分析2
泰坦尼克号乘客生存情况预测分析2
|
5月前
|
机器学习/深度学习 数据采集 JavaScript
技术心得记录:机器学习——用逻辑回归及随机森林实现泰坦尼克号的生存预测
技术心得记录:机器学习——用逻辑回归及随机森林实现泰坦尼克号的生存预测
52 0
|
6月前
|
机器学习/深度学习 安全 数据处理
泰坦尼克号乘客生存情况预测分析1
泰坦尼克号乘客生存情况预测分析1
|
6月前
|
机器学习/深度学习 算法 安全
泰坦尼克号乘客生存情况预测分析3
泰坦尼克号乘客生存情况预测分析3
|
6月前
|
存储
【视频】R语言中的分布滞后非线性模型(DLNM)与发病率,死亡率和空气污染示例
【视频】R语言中的分布滞后非线性模型(DLNM)与发病率,死亡率和空气污染示例
|
6月前
|
机器学习/深度学习 传感器 自然语言处理
时间序列预测的零样本学习是未来还是炒作:TimeGPT和TiDE的综合比较
最近时间序列预测预测领域的最新进展受到了各个领域(包括文本、图像和语音)成功开发基础模型的影响,例如文本(如ChatGPT)、文本到图像(如Midjourney)和文本到语音(如Eleven Labs)。这些模型的广泛采用导致了像TimeGPT[1]这样的模型的出现,这些模型利用了类似于它们在文本、图像和语音方面获得成功的方法和架构。
124 1
|
6月前
|
机器学习/深度学习 数据可视化 数据挖掘
SPSS用K均值聚类KMEANS、决策树、逻辑回归和T检验研究通勤出行交通方式选择的影响因素调查数据分析
SPSS用K均值聚类KMEANS、决策树、逻辑回归和T检验研究通勤出行交通方式选择的影响因素调查数据分析
|
6月前
|
机器学习/深度学习 算法 数据可视化
机器学习-生存分析:如何基于随机生存森林训练乳腺癌风险评估模型?
机器学习-生存分析:如何基于随机生存森林训练乳腺癌风险评估模型?
115 1
|
6月前
|
存储
R语言分布滞后非线性模型(DLNM)研究发病率,死亡率和空气污染示例
R语言分布滞后非线性模型(DLNM)研究发病率,死亡率和空气污染示例
|
6月前
|
存储
R语言分布滞后非线性模型(DLNM)空气污染研究温度对死亡率影响建模应用
R语言分布滞后非线性模型(DLNM)空气污染研究温度对死亡率影响建模应用