From 7b81f8241b14ec564cb11228abc9b7dee69acb91 Mon Sep 17 00:00:00 2001 From: Daniel Julius Lasiman Date: Mon, 29 May 2017 11:59:55 +0700 Subject: [PATCH] Fix lastfm example: Don't ignore the number of iterations --- examples/lastfm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/lastfm.py b/examples/lastfm.py index 639b2b27..07ef0cc2 100644 --- 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")