Skip to content

Commit

Permalink
set iteratios to 15
Browse files Browse the repository at this point in the history
  • Loading branch information
jasonzhang929 committed May 7, 2019
1 parent 9a4157e commit a078fd9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion SGD/gradient1.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def sparse_SGD(T, U, V, W, Lambda, omega, I, J, K, r, stepSize, sample_rate):
print(diff_norm, x, total_count, x/total_count)
# print(times)

if abs(curr_err_norm - next_err_norm) < .001 or iteration_count > work_cycle * 0:
if abs(curr_err_norm - next_err_norm) < .001 or iteration_count > work_cycle * 15:
break

curr_err_norm = next_err_norm
Expand Down

0 comments on commit a078fd9

Please sign in to comment.