决策树解决泰坦尼克号人员生存预测问题报告

从python和机器学习小白，大概是用了几天学了简单python，然后写了一个最简单的决策树算法，大部分都是调用,简单总结一些解决的泰坦尼克号问题。

首先是流程分析：1、获取数据。2、数据处理（特征值处理，特征值->字典类型）。3、划分数据集。4、特征工程（字典特征提取）。5、决策树预估器流程。6、模型评估。7、决策树可视化。

1、获取数据：

官网下载一直出现错误，所以我从CSDN论坛上面下载了一个，不过数据集有些问题，有一万多条数据，和事件情况不符，而且里面不少缺失数据，但是我看到很多人都是用这个数据集进行的分析。

代码：

2、数据处理：

我打开数据集发现影响生存率的条件很多，包括年龄、舱位等等。其中我在特征值中选出了年龄、性别、是不是贵族，年龄对缺失值进行了补充，原则是补上平均值。最后一步转换成字典。

代码：

3、划分数据集:

将训练集和测试集按照3：7分开，随机样本数设成50。

代码：