| |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
-> 人工智能 -> 【Pytorch】机器学习入门实践 以Titanic为例 -> 正文阅读 |
|
[人工智能]【Pytorch】机器学习入门实践 以Titanic为例 |
前言 最近学习了B站刘二大人的机器学习教程,传送门PyTorch深度学习实践——加载数据集。 课后习题为在Kaggle上找机器学习(ML)项目练手,Titanic问题是官方的入门项目,在此做一个记录。 一、处理数据集 在此次竞赛的目标是给定乘客的一些信息,使用机器学习创建一个模型,预测哪些乘客在泰坦尼克号沉船中幸存下来,即预测Survival为0还是1。这里Kaggle提供了三个数据集:test.csv是用来测试模型时所使用的数据集;train.csv是训练模型时所使用的数据集,里面带有标签;gender_submission.csv是上传文件范例,即我们需要按着这里面要求的数据格式将模型训练得到的结果上传到Kaggle上去验证。 ? ? 打开train.csv,查看里面的数据格式,依据官方给出的说明,确定各列的含义。
首先需要对特征进行一定的筛选,有一些特征我认为是对分类没有帮助或者不好处理的,比如Name、Ticket、Cabin,于是直接舍弃这几项特征,直接对剩下的特征进行建模,这里我选择了Pclass、Sex、Age、SibSp、Parch、Embarked等六个特征作为初始的targets,Survived作为初始的labels,同时将含有空数据的行剔除,共得到712行有效行(list)。
二、使用pytorch构建模型训练使用最简单的全连接层+激活函数的配置,从torch.nn中继承module类,构建模型如下:
训练也很简单,先设置好优化器和loss函数,然后读取数据进行正向传播,最后进行反向梯度传播并更新权重就可以了:
部分训练结果如下,因为特征构建与模型都很简单,所以没有太高的准确率。
三. 上传测试结果用训练好的模型在测试集上进行预测
得到官方要求格式的预测文件后,就可以去Kaggle官网上传结果了,在比赛主页点击Submit Predictions进入上传页面后,上传得到的csv文件即可。 上传完毕后会给一个评分,点击评分下面的蓝色链接会给你看你的排名数据,我这个差不多排到一万八千多名以后了,看排名靠前的都有100%的预测准确率,说明对于这个任务,不管是在特征工程还是在模型方面都还有很大的上升空间的。 结束语整个流程就这样结束了,这也算是我在Kaggle上弄的第一个项目,主要是为了弄清楚其上面的竞赛流程并做一个记录。不得不说这是一个很好的锻炼自己机器学习编码能力与算法设计能力的平台,以后也希望自己能在上面多花一些时间做做竞赛。 |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 | -2024/11/27 10:43:25- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |