Spaces:
Running
Running
Update app.py
Browse files
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
|