Skip to content

Commit 2d98c34

Browse files
Update softmax_regression_train.py
1 parent 775d96e commit 2d98c34

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Chapter_2 Softmax Regression/softmax_regression_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def save_model(file_name, weights):
8989
feature, label, k = load_data(inputfile)
9090
# 2、训练Softmax模型
9191
print "---------- 2.training ------------"
92-
weights = gradientAscent(feature, label, k, 5000, 0.2)
92+
weights = gradientAscent(feature, label, k, 10000, 0.4)
9393
# 3、保存最终的模型
9494
print "---------- 3.save model ------------"
9595
save_model("weights", weights)

0 commit comments

Comments
 (0)