From d38c3f72ee706de148f8e622018331d9d4a11ddb Mon Sep 17 00:00:00 2001 From: xuyao Date: Sun, 14 Jan 2024 17:35:42 +0800 Subject: [PATCH] fix bugs in sparql and eval --- ToG/freebase_func.py | 2 +- eval/utils.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/ToG/freebase_func.py b/ToG/freebase_func.py index e141764..f17d241 100644 --- a/ToG/freebase_func.py +++ b/ToG/freebase_func.py @@ -154,7 +154,7 @@ def entity_search(entity, relation, head=True): tail_entities_extract = sparql_tail_entities_extract% (entity, relation) entities = execurte_sparql(tail_entities_extract) else: - head_entities_extract = sparql_head_entities_extract% (entity, relation) + head_entities_extract = sparql_head_entities_extract% (relation, entity) entities = execurte_sparql(head_entities_extract) diff --git a/eval/utils.py b/eval/utils.py index fdcd977..883cbb6 100644 --- a/eval/utils.py +++ b/eval/utils.py @@ -104,6 +104,8 @@ def check_string(string): return "{" in string def clean_results(string): + if string.lower().startswith('{yes}'): + string = string[5:] if "{" in string: start = string.find("{") + 1 end = string.find("}")