Fix/Recommendation

wafflestudio · Feb 2, 2025 · 06ea384 · 06ea384
1 parent a007540
commit 06ea384
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/watchapedia/app/recommend/service.py b/watchapedia/app/recommend/service.py
@@ -66,7 +66,7 @@ def get_expected_rating(self, user_id: int):
 
         movie_list = self.movie_service.search_movie_list("")
         movie_list.sort(key = lambda movie : movie.reviews_count, reverse=True)
-        movie_list = movie_list[0:18]
+        movie_list = movie_list[0:10]
 
         user_average = self.user_average_rating(user_id)
 
@@ -86,7 +86,7 @@ def get_expected_rating(self, user_id: int):
         expect_num = 0
 
         for movie in movie_list :
-            if expect_num >= 8 :
+            if expect_num >= 7 :
                 break
 
             if movie.average_rating is None :