Update Recommender.py

Recommender.py

@@ -3,9 +3,11 @@ import pandas as pd
 from sklearn.metrics.pairwise import cosine_similarity
 
 class Recommender :
-    def __init__(self, title_vec_col, content_vec_col, id_col):
+    def __init__(self, id_col, title_col, content_col, title_vec_col, content_vec_col):
         self.title_vec_col = title_vec_col
         self.content_vec_col = content_vec_col
+        self.title_col = title_col
+        self.content_col = content_col
         self.id_col = id_col
 
     def calculate_recom_scores (self, k, similarities) :
@@ -56,10 +58,10 @@ class Recommender :
 
     #     return result[self.id_col, :].tolist()
 
-    def recommend_k(self, table, k, id):
+    def recommend_k(self, table, k, title):
 
       data = np.array(list(zip(*table)))
-      idx = int(data[0][data[self.id_col] == id].item())
+      idx = int(data[0][data[self.title_col] == title].item())
 
       titles = self.str2arr(data[self.title_vec_col, :])
       contents = self.str2arr(data[self.content_vec_col, :])