PyTorch的4分钟教程，手把手教你完成线性回归(3)

发布时间：2019-07-29 00:20 所属栏目：19 来源：大数据文摘

导读：运行结果： MSELoss参考网站：https://pytorch.org/docs/stable/_modules/torch/nn/modules/loss.html SGD参考网站：https://pytorch.org/docs/stable/_modules/torch/optim/sgd.html 第四步：现在训练已经完成，

运行结果：

PyTorch的4分钟教程，手把手教你完成线性回归

MSELoss参考网站：https://pytorch.org/docs/stable/_modules/torch/nn/modules/loss.html
SGD参考网站：https://pytorch.org/docs/stable/_modules/torch/optim/sgd.html

第四步：

现在训练已经完成，让我们直观地检查我们的模型：

#Step 4:Display model and confirm 
import matplotlib.pyplot as plt 
plt.figure(figsize=(4,4)) 
plt.title("Model and Dataset") 
plt.xlabel("X");plt.ylabel("Y") 
plt.grid() 
plt.plot(x,y,"ro",label="DataSet",marker="x",markersize=4) 
plt.plot(x,model.model.weight.item()*x+model.model.bias.item(),label="Regression Model") 
plt.legend();plt.show()

运行结果：

PyTorch的4分钟教程，手把手教你完成线性回归

现在你已经完成了PyTorch的第一个线性回归例子的编程了，对于后续希望百尺竿头，更进一步的读者来说，可以参考PyTorch的官方文档链接，完成大部分的编码应用。