第六讲 - 逻辑斯蒂回归【跟随 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 的小伙伴可以 私信我 进群哦。
以上就是我要分享的内容,因为学识尚浅,会有不足,还请各位大佬指正。 有什么问题也可在评论区留言。
|