diff --git a/abl/abducer/kb.py b/abl/abducer/kb.py index 7aff683..71bdcbe 100644 --- a/abl/abducer/kb.py +++ b/abl/abducer/kb.py @@ -282,9 +282,7 @@ class prolog_KB(KBBase): class add_prolog_KB(prolog_KB): def __init__(self, pseudo_label_list=list(range(10))): super().__init__(pseudo_label_list) - for i in self.pseudo_label_list: - self.prolog.assertz("pseudo_label(%s)" % i) - self.prolog.assertz("addition(Z1, Z2, Res) :- pseudo_label(Z1), pseudo_label(Z2), Res is Z1+Z2") + self.prolog.consult('../datasets/mnist_add/add.pl') class HED_prolog_KB(prolog_KB):