Update Linear_Regression.py

Author: liweiwei1419

01-Linear Regression/Linear_Regression.py

@@ -59,6 +59,8 @@ def forward(self, x):
 predict = predict.data.numpy()
 plt.plot(x_train.numpy(), y_train.numpy(), 'ro', label='Original data')
 plt.plot(x_train.numpy(), predict, label='Fitting Line')
+# 显示图例
+plt.legend() 
 plt.show()
 
 # 保存模型