Update softmax_regression_train.py

zhaozhiyong19890102 · web-flow · commit 2d98c3406ac5 · 2017-10-14T22:30:50.000+08:00
diff --git a/Chapter_2 Softmax Regression/softmax_regression_train.py b/Chapter_2 Softmax Regression/softmax_regression_train.py
@@ -89,7 +89,7 @@ def save_model(file_name, weights):
 feature, label, k = load_data(inputfile)
 # 2、训练Softmax模型
 print "---------- 2.training ------------"
- weights = gradientAscent(feature, label, k, 5000, 0.2)
+ weights = gradientAscent(feature, label, k, 10000, 0.4)
 # 3、保存最终的模型
 print "---------- 3.save model ------------"
 save_model("weights", weights)

-Original file line number
+Diff line change
 feature, label, k=load_data(inputfile)
 # 2、训练Softmax模型
 print"---------- 2.training ------------"
 -weights=gradientAscent(feature, label, k, 5000, 0.2)
 +weights=gradientAscent(feature, label, k, 10000, 0.4)
 # 3、保存最终的模型
 print"---------- 3.save model ------------"
 save_model("weights", weights)