modify call to main

cyclops-community · May 7, 2019 · eee2244 · eee2244
1 parent 99cc86d
commit eee2244
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/ALS/als_sp.py b/ALS/als_sp.py
@@ -310,15 +310,15 @@ def getALS_CG(T,U,V,W,regParam,omega,I,J,K,r,block_size,num_iter=100,err_thresh=
 
         it += 1
 
-        if abs(curr_err_norm - next_err_norm) < err_thresh or it > num_iter:
+        if abs(curr_err_norm - next_err_norm) < err_thresh or it >= num_iter:
             break
 
         curr_err_norm = next_err_norm
 
     t_ALS_CG.end()
 
     if glob_comm.rank() == 0:
-        print('Time/Iteration: {}'.format((time.time() - t_before_loop)/1))
+        print('Time: {}'.format((time.time() - t_before_loop)/1))
         print("Number of iterations: %d" % (it))
 
 
@@ -348,9 +348,9 @@ def main():
     r = 2 
     sparsity = .000001
     regParam = .1
-    block_size = 100
+    block_size = 1000
     use_func = 0
-    num_iter = 20
+    num_iter = 1
     err_thresh = .001
     run_implicit = 1
     run_explicit = 1
@@ -416,5 +416,7 @@ def main():
 
 
 
-main()
+#main()
+if __name__ == '__main__':
+    main()
 
diff --git a/SGD/gradient1.py b/SGD/gradient1.py
@@ -163,4 +163,4 @@ def main():
     #T.write_to_file("T.txt")
     sparse_SGD(T, U, V, W, regParam, Omega, I, J, K, r, stepSize, sample_rate)
 
-main()
+#main()
-Original file line number
+Diff line change
@@ Expand Up / @@ -163,4 +163,4 @@ def main(): @@
         #T.write_to_file("T.txt")
         sparse_SGD(T, U, V, W, regParam, Omega, I, J, K, r, stepSize, sample_rate)
-    main()
+    #main()