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

Merge branch 'dev' of https://github.com/xerrors/ProjectAthena into dev

Browse Source
This commit is contained in:
Wenjie Zhang 2024-08-02 18:47:40 +08:00
commit 786a02eb6a
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -34,3 +34,4 @@ neo4j*
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])