概要

主要利用Matlab的分类学习器快速搭建机器学习模型,预测Kaggle入门题目泰坦尼克号成员存活概率。

数据说明

train.csv包含了船上部分乘客(891名乘客,每位乘客在表格中都有不同的一行)的详细信息。数据如下图所示。
数据详情

第二列(“Survived”)中的值可用于确定每位乘客是否存活:

如果是 “1”,则乘客存活。
如果是 “0”,则乘客死亡。

数据导入

数据导入

利用Matlab的数据导入功能将训练数据导入至工作区域,对变量的类型以及一些缺失值进行处理
分类学习器
打开Matlab分类学习器,将相关的训练数据导入,选择需要采用的特征,这里只是简单的演示,将名字等信息直接剔除掉,利用简单的信息进行预测测试

模型训练

模型训练

点击训练按钮之后,即可选择相关的模型进行训练,训练好的模型可以进行导出,以便后续进行调用

小结

预测模型

这里选取的特征未经过任何预处理,可以进行处理,将一些缺失值进行填充,进一步优化模型。
相关训练代码与数据导入函数链接

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐