引言
这部分可以说是独立于我们整个项目的一小节,不过也是极为重要的一小节,这一节的学习可以帮助我们去了解神经网络。我们用一个很简单的例子进行参考,本节内容的知识参考了莫烦python的系列教程。
https://space.bilibili.com/243821484?spm_id_from=333.788.b_765f7570696e666f.1
好了,话不多说,我们进入学习。
正文
我们直接开始代码的编写
这个是单独的项目,所以大家可以跟着一步一步来完成并实现这个代码。
首先我们包含一下可能会用到的头文件
import torch
from torch.autograd import Variable
import torch.nn.functional as F
# 导入plot模块需要以下三句
import matplotlib.pyplot as plt
import matplotlib; matplotlib.use('TkAgg')
import os;os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
因为我用的pycharm,当时写代码的时候因为plot没法弹窗,后来查完发现按上述方法写就可以了。
我们继续正题,我们下面来造一组伪随机点,代码如下
x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
y = x.pow(2) + 0.2 * torch.rand(x.size())
因为在torch中,数据