Skip to content

Commit 22a0e8c

Browse files
Update ridge_regression_train.py
1 parent aa91641 commit 22a0e8c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Chapter_8 RidgeRegression/ridge_regression_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def save_weights(file_name, w0):
208208
if method == "bfgs": # 选择BFGS训练模型
209209
w0 = bfgs(feature, label, 0.5, 1000)
210210
elif method == "lbfgs": # 选择L-BFGS训练模型
211-
w0 = lbfgs(feature, label, 0.5, 10, m=10)
211+
w0 = lbfgs(feature, label, 0.5, 1000, m=10)
212212
else: # 使用最小二乘的方法
213213
w0 = ridge_regression(feature, label, 0.5)
214214
# 3、保存最终的模型

0 commit comments

Comments
 (0)