diff --git a/abl/learning/abl_model.py b/abl/learning/abl_model.py index f1512fe..8c8b2f9 100644 --- a/abl/learning/abl_model.py +++ b/abl/learning/abl_model.py @@ -10,7 +10,7 @@ # # ================================================================# import pickle -from utils import flatten, reform_idx +from ..utils import flatten, reform_idx from typing import List, Any, Optional diff --git a/abl/reasoning/kb.py b/abl/reasoning/kb.py index b700aa1..b551df0 100644 --- a/abl/reasoning/kb.py +++ b/abl/reasoning/kb.py @@ -142,7 +142,7 @@ class ground_KB(KBBase): key_idx = bisect.bisect_left(key_list, y) all_candidates = [] - for idx in range(key_idx - 1, 0, -1): + for idx in range(key_idx - 1, -1, -1): k = key_list[idx] if abs(k - y) <= self.max_err: all_candidates.extend(potential_candidates[k]) diff --git a/examples/hed/hed_example.ipynb b/examples/hed/hed_example.ipynb index 2e5add3..71830bd 100644 --- a/examples/hed/hed_example.ipynb +++ b/examples/hed/hed_example.ipynb @@ -12,7 +12,7 @@ "\n", "from abl.reasoning import ReasonerBase, prolog_KB\n", "from abl.learning import BasicNN, ABLModel\n", - "from abl.evaluation import SymbolMetric, ABLMetric\n", + "from abl.evaluation import SymbolMetric, SemanticsMetric\n", "from abl.utils import ABLLogger, reform_idx\n", "\n", "from examples.hed.hed_bridge import HEDBridge\n", @@ -206,7 +206,7 @@ "outputs": [], "source": [ "# Add metric\n", - "metric = [SymbolMetric(prefix=\"hed\"), ABLMetric(prefix=\"hed\")]" + "metric = [SymbolMetric(prefix=\"hed\"), SemanticsMetric(prefix=\"hed\")]" ] }, {