Spaces:
Sleeping
Sleeping
Commit
·
6d975f0
1
Parent(s):
ba0b88d
Update Recommender.py
Browse files- Recommender.py +1 -1
Recommender.py
CHANGED
@@ -62,7 +62,7 @@ class Recommender :
|
|
62 |
def recommend_k(self, table, k, title):
|
63 |
|
64 |
data = np.array(list(zip(*table)))
|
65 |
-
idx =
|
66 |
titles = self.str2arr(data[self.title_vec_col, :])
|
67 |
contents = self.str2arr(data[self.content_vec_col, :])
|
68 |
|
|
|
62 |
def recommend_k(self, table, k, title):
|
63 |
|
64 |
data = np.array(list(zip(*table)))
|
65 |
+
idx = np.where(data == title)[0].tolist()[0]
|
66 |
titles = self.str2arr(data[self.title_vec_col, :])
|
67 |
contents = self.str2arr(data[self.content_vec_col, :])
|
68 |
|