实例:泰坦尼克号幸存者的预测

本文使用分类树模型预测泰坦尼克号乘客的生存可能性,分析数据集并探索相关变量,通过填充缺失值和相关检验优化模型。最终,利用网格搜索调整参数,评估模型效果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

泰坦尼克号的沉没是世界上最严重的海难事故之一,今天我们通过分 类树模型来预测一下哪些人可能成为幸存者。数据集来着https://www.kaggle.com/c/titanic,数据集会随着代码一起提供给大家,大家可以在下载页面拿到,或者到群中询问。数据集包含两个csv格式文件,data为我们接下来要使用的数据,test为kaggle提供的测试集。

  • 导入所需要的库
  • 导入数据集,探索数据
  • 相关变量分析
    *查看存活率
    *单变量之间的分析
    *多变量之间的分析
    *缺失值的填充
    *相关检验
  • 对数据集进行不涉及到训练集和测试集之间相互影响的预处理(分箱)
  • 提取标签和特征矩阵,分测试集和训练集
  • 导入模型
  • 观察模型的拟合状况
  • 调参数(我们这里用网格格搜索调整参数)
  • 评估模型,ROC曲线
    具体代码如下:https://nbviewer.jupyter.org/github/xiaotaiyang08/123/blob/master/练习案例泰坦尼克.ipynb
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值