Skip to content

Commit 0bbb180

Browse files
committed
update to grid layout & fix error in duplicate finder
1 parent 0ab2fc4 commit 0bbb180

File tree

4 files changed

+462
-469
lines changed

4 files changed

+462
-469
lines changed

efficient_ir.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,6 @@ def add_fv(self, fv, idx):
7171

7272

7373
def match(self, fv, nc=5):
74-
similarity = (1-np.tanh(self.hnsw_index.knn_query(fv, k=nc)[1][0]/3000))*100
75-
results = self.hnsw_index.knn_query(fv, k=nc)[0][0]
76-
return similarity, results
74+
query = self.hnsw_index.knn_query(fv, k=nc)
75+
similarity = (1-np.tanh(query[1][0]/3000))*100
76+
return similarity, query[0][0]

gui.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ def _init_ui_(self):
3939
self.resultTableDuplicate.horizontalHeader().setSectionResizeMode(0, QtWidgets.QHeaderView.Stretch)
4040
self.resultTableDuplicate.horizontalHeader().setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
4141
self.resultTableDuplicate.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
42+
self.resultTableDuplicate.setSortingEnabled(True)
4243
self.searchDirTable.horizontalHeader().setSectionResizeMode(QtWidgets.QHeaderView.Stretch)
4344
self.searchDirTable.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
4445
self.update_dir_table()

0 commit comments

Comments
 (0)