Pytorch深度学习(一):前馈神经网络(FNN)
参考B站课程:《PyTorch深度学习实践》完结合集
传送门:《PyTorch深度学习实践》完结合集
一、线性模型:
已知数据:
x = [ 1 , 2 , 3 ] , y = [ 2 , 4 , 6 ] x=[1,2,3],\quad y=[2,4,6] x=[1,2,3],y=[2,4,6]
预测 x = 4 x=4 x=4时, y y y等于多少
建立线性模型
y ^ = w ∗ x , x , y ^ ∈ R \hat{y}=w*x,\quad x,\hat{y}\in\mathbb{R} y^=w∗x,x,y^∈R
估计权重 w w w(weight),再来预测 x = 4 x=4 x=4时 y y y的值。
定义MSE(Mean Square Error):
c o s t = 1 N ∑ n = 1 N ( y ^ n − y n ) 2 cost=\frac{1}{N}\sum_{n=1}^{N} (\hat{y}_n-y_n)^2 cost=N1n=1∑N(y^n