第六讲 - 逻辑斯蒂回归【跟随 up 主 “刘二大人” 学习 pytorch】

前言

• 本专栏是我这个小菜鸡跟随 B 站 up 主 刘二大人 学习 pytorch 完成的课后作业，原视频请戳这里
• 偷偷说一句：如果对我的文章满意的话可不可以给我 点个赞 ， 点个收藏， 点个关注，评论一下。
• 一定要 看到最后 并且参加 投票 哦。
• Let's do it !!!

下载数据集

既然是要进行 深度学习 ，数据集 必不可少，不管是 训练集 ，还是 测试集 ，亦或者是 验证集。PyTorch 中就有个工具包 torchvision ，这个包中的 datasets 为我们提供了下载一些 流行的数据集 ，比如说 MNIST （手写图片识别），CIFAR （图像分类）之类的。

torchvision 使用方法

看一下 官方文档 。我们可以下载这些数据集。

以 MNIST 为例，要使用 MNIST 数据集的话代码如下：

torchvision.datasets.MNIST(root='./training', train=True, download=True)

其中：

• root， 指定 数据集文件路径。如果有数据集填入 数据集的本地路径，如果没有，则填入 下载数据集的保存路径。
• train， 指定下载的 数据集 的 类型。True 代表 训练集，False 代表 测试集。
• dowmload，指定 是否下载数据集。True 则 从网上下载，False 则 不下载 ，使用 本地数据集。

下载完后大概是这个样子。

然后查看自己指定的路径，就可以找到咯。
其他的数据集也类似。下载速度的话，就只用看 自己的网速。

逻辑斯蒂回归

这一讲大佬 up 🐖 讲了 逻辑斯蒂回归。总结可以去看 我的这篇文章。

---

---

---

结尾

有想要一起学习 python 的小伙伴可以 私信我 进群哦。

以上就是我要分享的内容，因为学识尚浅，会有不足，还请各位大佬指正。
有什么问题也可在评论区留言。