Diangle commited on
Commit
c5eb030
·
1 Parent(s): 7fe2138

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -111,6 +111,7 @@ class NearestNeighbors:
111
  rerank_search = NearestNeighbors(n_neighbors=self.n_neighbors, metric='cosine')
112
  rerank_search.fit(rerank_data)
113
  re_sim, re_idx = rerank_search.kneighbors(np.asarray([q]))
 
114
  re_sims[i, :] = re_sim
115
  re_idxs[i, :] = idx[i][re_idx]
116
  idx = re_idxs
 
111
  rerank_search = NearestNeighbors(n_neighbors=self.n_neighbors, metric='cosine')
112
  rerank_search.fit(rerank_data)
113
  re_sim, re_idx = rerank_search.kneighbors(np.asarray([q]))
114
+ print("re_idx: ", re_idx)
115
  re_sims[i, :] = re_sim
116
  re_idxs[i, :] = idx[i][re_idx]
117
  idx = re_idxs