From d5fe9eaced42c65378494a3807e7e9a8b3bfd5a4 Mon Sep 17 00:00:00 2001 From: ahmed531998 Date: Tue, 18 Apr 2023 13:13:36 +0200 Subject: [PATCH] fix --- Recommender.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/Recommender.py b/Recommender.py index 2d99aa2..9b13085 100644 --- a/Recommender.py +++ b/Recommender.py @@ -23,9 +23,13 @@ class Recommender: if cosine_similarity(np.array(self.retriever.encode([tag])), np.array(self.retriever.encode([interest]))) > score: if self._new(username, material): - print('hi') - self.curr_recommendations[username] = self.curr_recommendations[username].append(material) if username not in self.curr_recommendations else [material] - self.recommended[username] = self.recommended[username].append(False) if username not in self.recommended else [False] + #print('hi') + if username in self.curr_recommendations: + self.curr_recommendations[username].append(material) + self.recommended[username].append(False) + else: + self.curr_recommendations[username] = [material] + self.recommended[username] = [False] def generate_recommendations(self, username, interests, new_material): for interest in interests: