用决策树分析泰坦尼克号生存

一）前言
上一篇文章将了决策树分类的原理以及参数的分析，这篇就用用决策树分析泰坦尼克号生存情况。
上一篇地址：
https://blog.youkuaiyun.com/MR_Trustin/article/details/96886157

二）代码以及分析
数据来源于kaggle，地址是https://www.kaggle.com/c/titanic

#导入相关模块
%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
import sklearn.metrics as sm

data = pd.read_csv('titanic/train.csv',index_col=0) #指定第一列为索引
# 观察数据结构
data.head()

# 通过观察，丢弃无用的数据'Name', 'Ticket', 'Cabin'
data = data.drop(['Name', 'Ticket', 'Cabin'],axis=1)
data.head()
</