回归树练习，泰坦尼克号幸存者的预测

原创

已于 2024-10-16 16:21:59 修改 · 311 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#回归 #数据挖掘 #人工智能

于 2024-10-05 10:12:40 首次发布

回归树练习，泰坦尼克号幸存者的预测

数据集下载地址
https://download.youkuaiyun.com/download/AnalogElectronic/89846327
包含测试集合代码资源
https://download.youkuaiyun.com/download/AnalogElectronic/89892984

我们来看看train.csv文件，它包含了891个样本，每个样本代表一个乘客。这些样本的数据包括乘客的年龄（Age）、船票等级（Pclass）、性别（Sex）、登船港口（Embarked）、票价（Fare）等基本信息，以及最重要的生存状态（Survived）。这些特征提供了对乘客生存可能性的洞察，比如男性与女性的生存率差异、船票等级与生存机会的关系等。

##回归树练习，泰坦尼克号幸存者的预测
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split
from sklearn.model_selection import GridSearchCV
from sklearn.model_selection import cross_val_score
import matplotlib.pyplot as plt
data = pd.read_csv(r"I:\hadoop note\titanic_train.csv",index_col= 0)
data.head()

在这里插入图片描述


#删除缺失值过多的列，和观察判断来说和预测的y没有关系的列
data.drop(["Cabin","Name","Ticket"],inplace=True,axis=1)
#处理缺失值，对缺失值较多的列进行填补，有一些特征只确实一两个值，可以采取直接删除记录的方法
data["Age"] = data["Age"].fillna(data["Age"].mean())
data = data.dropna()
#将分类变量转换为数值型变量
#将二分类变量转换为数值型变量
#astype能够将一个pandas对象转换为某种类型，和apply(int(x))不同，astype可以将文本类转换为数字，用这个方式可以很便捷地将二分类特征转换为0~1
data["Sex"] = (data["Sex"]== "male").astype("int")
#将三分类变量转换为数值型变量
labels = data["Embarked"].unique().tolist()
data["Embarked"] = data["Embarked"].apply(lambda x: labels.index(x))
#查看处理后的数据集
data.head()

在这里插入图片描述

##提取X和Y，拆分训练集和测试集
X = data.iloc[:,data.columns != "Survived"]
y = data.iloc[:,data.columns == "Survived"]
from sklearn.model_selection import train_test_split
Xtrain, Xtest, Ytrain, Ytest = train_test_split(X,y,test_size=0.3)
#修正测试集和训练集的索引
for i in [Xtrain, Xtest, Ytrain, Ytest]:
    i.index = range