amirhosseinkarami commited on
Commit
6d975f0
·
1 Parent(s): ba0b88d

Update Recommender.py

Browse files
Files changed (1) hide show
  1. 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 = int(data[0][data[self.title_col] == title].tolist()[0])
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