IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: 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,查看里面的数据格式,依据官方给出的说明,确定各列的含义。

  • PassengerId:乘客ID,是乘客的唯一标示符
  • Survived:乘客是否存活,相当于标签
  • Pclass:乘客坐的是几等座,分为1,2,3三个等级
  • Name:乘客的名字
  • Sex:乘客的性别,分为female/male
  • Age:乘客的年龄,需要注意的是这列数据中有缺失值
  • SibSp:乘客的兄弟/配偶在船上的数量
  • Parch:乘客的父母/子女在船上的数量
  • Ticket:乘客船票的编号
  • Fare:乘客船票的价格
  • Cabin:乘客的房间号
  • Embarked:乘客的出发港,分为C、Q、S三个

首先需要对特征进行一定的筛选,有一些特征我认为是对分类没有帮助或者不好处理的,比如Name、Ticket、Cabin,于是直接舍弃这几项特征,直接对剩下的特征进行建模,这里我选择了Pclass、Sex、Age、SibSp、Parch、Embarked等六个特征作为初始的targets,Survived作为初始的labels,同时将含有空数据的行剔除,共得到712行有效行(list)。

targets = torch.Tensor(list[:,:-1])
labels = torch.Tensor(list[:,[-1]])

二、使用pytorch构建模型训练

使用最简单的全连接层+激活函数的配置,从torch.nn中继承module类,构建模型如下:

class Model(nn.Module):

    def __init__(self):
        super(Model, self).__init__()
        self.linear1 = nn.Linear(6,4)
        self.linear2 = nn.Linear(4,2)
        self.linear3 = nn.Linear(2,1)
        self.sigmod = nn.Sigmod()

    def forward(self,x):
        x = self.sigmod(self.linear1(x))
        x = self.sigmod(self.linear2(x))
        x = self.sigmod(self.linear3(x))
        return x

训练也很简单,先设置好优化器和loss函数,然后读取数据进行正向传播,最后进行反向梯度传播并更新权重就可以了:

criterion = nn.BCELoss()
optimizer = torch.optim.Adam(model.parameters(),lr = 0.1)

for epoch in range(100000):
    pred = model(targets)
    # print(pred.size())
    loss = criterion(pred,labels)
    print(epoch,loss.item())

    optimizer.zero_grad()
    loss.backward()
    optimizer.step()

部分训练结果如下,因为特征构建与模型都很简单,所以没有太高的准确率。

三. 上传测试结果

用训练好的模型在测试集上进行预测

y_pred = model(targets_test)
y_pred = torch.where(y_pred>0.5,torch.Tensor([1.]),torch.tensor([0.]))

得到官方要求格式的预测文件后,就可以去Kaggle官网上传结果了,在比赛主页点击Submit Predictions进入上传页面后,上传得到的csv文件即可。

上传完毕后会给一个评分,点击评分下面的蓝色链接会给你看你的排名数据,我这个差不多排到一万八千多名以后了,看排名靠前的都有100%的预测准确率,说明对于这个任务,不管是在特征工程还是在模型方面都还有很大的上升空间的。

结束语

整个流程就这样结束了,这也算是我在Kaggle上弄的第一个项目,主要是为了弄清楚其上面的竞赛流程并做一个记录。不得不说这是一个很好的锻炼自己机器学习编码能力与算法设计能力的平台,以后也希望自己能在上面多花一些时间做做竞赛。

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2021-10-07 13:51:15  更:2021-10-07 13:53:44 
 
开发: 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年5日历 -2024/5/21 20:07:41-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码