Merge pull request #32 from danieljl/fix-lastfm-example

Fix lastfm example: Don't ignore the number of iterations
benfred · May 29, 2017 · c2d3aaf · c2d3aaf
2 parents e6ff9fc + 7b81f82
commit c2d3aaf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/lastfm.py b/examples/lastfm.py
@@ -67,11 +67,11 @@ def calculate_similar_artists(input_filename, output_filename,
         if exact:
             model = AlternatingLeastSquares(factors=factors, regularization=regularization,
                                             use_native=use_native, use_cg=cg,
-                                            dtype=dtype)
+                                            dtype=dtype, iterations=iterations)
         else:
             model = AnnoyAlternatingLeastSquares(factors=factors, regularization=regularization,
                                                  use_native=use_native, use_cg=cg,
-                                                 dtype=dtype)
+                                                 dtype=dtype, iterations=iterations)
 
         # lets weight these models by bm25weight.
         logging.debug("weighting matrix by bm25_weight")