From 54e4dc77ab384cb8506ed0e8b491a7e9b3a41cf4 Mon Sep 17 00:00:00 2001 From: Xi Xu Date: Wed, 16 Oct 2024 20:07:58 +0800 Subject: [PATCH] Update LinearRegression_1.py --- 03_linear_regression/LinearRegression_1.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/03_linear_regression/LinearRegression_1.py b/03_linear_regression/LinearRegression_1.py index 41acf0b..f7e67d0 100644 --- a/03_linear_regression/LinearRegression_1.py +++ b/03_linear_regression/LinearRegression_1.py @@ -158,3 +158,7 @@ if __name__ == "__main__": gd2 = LinearRegressionGD2(hp) gd2.solve(lr, num_epochs) print(f"next prediction year (GD2): {gd2.linout([len(hp)])}") + + lr = 0.03 + gd1.solve(lr, num_epochs) + print(f"next prediction year (GD1, learning rate 0.03): {gd1.linout([len(hp)])}")