diff --git a/algo/adreco.py b/algo/adreco.py index 142e3b22cac41a03f8a10fc9c89eaf9d100d608c..a3abc0ff319b20c788fa1bed2a321ef68b86f1db 100644 --- a/algo/adreco.py +++ b/algo/adreco.py @@ -204,7 +204,7 @@ def user_profile( user_index, moviesdf, usersdf, vectMatrix ): similarity_scores_sorted = sorted(similarity_scores, key=lambda x: x[1], reverse=True) #lists recommendations index of the movies, ordered by weights - recommendations_indices = [t[0] for t in similarity_scores_sorted[1:(100+1)]] + recommendations_indices = [t[0] for t in similarity_scores_sorted[1:(30+1)]] return recommendations_indices @@ -238,11 +238,11 @@ def updateDB(): rec_indices=user_profile( i, moviesdf, usersdf, vect_matrix) if rec_indices != None: - recdf = moviesdf['id'].iloc[rec_indices] + recdf = moviesdf['_id'].iloc[rec_indices] titledf = moviesdf['original_title'].iloc[rec_indices] for j in recdf.index: - recommended_movies.append(int(recdf[j])) + recommended_movies.append(recdf[j]) dict['recommended_movies']=recommended_movies