Skip to content

Commit 9900a76

Browse files
authored
Merge pull request L1aoXingyu#4 from weimingge14/patch-1
Update Linear_Regression.py
2 parents d2192f5 + f2f0506 commit 9900a76

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

01-Linear Regression/Linear_Regression.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,8 @@ def forward(self, x):
5959
predict = predict.data.numpy()
6060
plt.plot(x_train.numpy(), y_train.numpy(), 'ro', label='Original data')
6161
plt.plot(x_train.numpy(), predict, label='Fitting Line')
62+
# 显示图例
63+
plt.legend()
6264
plt.show()
6365

6466
# 保存模型

0 commit comments

Comments
 (0)