kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

图检索中的query_by_vector小bug

Browse Source
This commit is contained in:
QiyiJiang 2024-07-31 14:50:46 +00:00
parent cb00a8f4a9
commit d5121b446a
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -33,4 +33,5 @@ neo4j*
*/package-lock.json */package-lock.json
web/package-lock.json web/package-lock.json
saves saves
notebooks notebooks
*.yaml

4
src/core/graphbase.py
View File

@ -277,10 +277,12 @@ class GraphDatabase:
with self.driver.session() as session: with self.driver.session() as session:
return session.execute_read(query, keyword) return session.execute_read(query, keyword)
def query_by_vector(self, entity_name, num_of_res=2, threshold=0.9, kgdb_name='neo4j', hops=2): def query_by_vector(self, entity_name, threshold=0.9,kgdb_name='neo4j', hops=2, num_of_res=2):
self.use_database(kgdb_name) self.use_database(kgdb_name)
result = self.query_by_vector_tep(entity_name) result = self.query_by_vector_tep(entity_name)
querys = [] querys = []
threshold = 0.9 if threshold is None else threshold
num_of_res = 2 if num_of_res is None else num_of_res
for i in range(num_of_res): for i in range(num_of_res):
if result[i][1] > threshold: if result[i][1] > threshold:
querys.append(result[i][0]) querys.append(result[i][0])