From 1d8d1eb0091f993d2a6b118ba3f3111ae81b969d Mon Sep 17 00:00:00 2001
From: Tom Bray <tom.bray@student-cs.fr>
Date: Fri, 10 Jun 2022 13:28:22 +0200
Subject: [PATCH] modification de l'id de l'algo
---
algo/adreco.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/algo/adreco.py b/algo/adreco.py
index 142e3b2..a3abc0f 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
--
GitLab