def disp_fn(eng, it, session, cost, outp):
mismatch_count = 0
bg = all_bgs[args.TEST_SET_INDEX]
mask = bg.get_target_mask('SameBib')
true_class = bg.get_target_data('SameBib')
pred_class = outp['SameBib'][0, :]
true_class = true_class[mask > 0]
pred_class = pred_class[mask > 0]
avg_acc = average_precision_score(true_class, pred_class)
auroc = roc_auc_score(true_class, pred_class)
x, y, th1 = precision_recall_curve(true_class, pred_class)
aupr = auc(y, x)
acc = accuracy_score(true_class, (pred_class >= .5).astype(np.float))
print('--------------------------------------------------------')
print('acc : %.4f , avg acc : %4f, AUROC : %.4f, AUPR : %.4f' %
(acc, avg_acc, auroc, aupr))
return
model = ILPRLEngine(args=args, predColl=predColl, bgs=bgs, disp_fn=disp_fn)
model.train_model()
help='Per value accuracy threshold',
type=float)
parser.add_argument('--PLOGENT',
default=.50,
help='Crossentropy coefficient',
type=float)
parser.add_argument('--BETA1', default=.90, help='ADAM Beta1', type=float)
parser.add_argument('--BETA2', default=.999, help='ADAM Beta2', type=float)
parser.add_argument('--EPS', default=1e-6, help='ADAM Epsillon', type=float)
parser.add_argument('--GPU', default=1, help='Use GPU', type=int)
parser.add_argument('--LOGDIR',
default='./logs/Logic',
help='Log Dir',
type=str)
parser.add_argument('--TB', default=0, help='Use Tensorboard', type=int)
parser.add_argument('--SEED', default=0, help='Random seed', type=int)
parser.add_argument('--ADDGRAPH',
default=1,
help='Add graph to Tensorboard',
type=int)
parser.add_argument('--CLIP_NORM', default=0, help='Clip gradient', type=float)
args = parser.parse_args()
print('displaying config setting...')
for arg in vars(args):
print('{}-{}'.format(arg, getattr(args, arg)))
model = ILPRLEngine(args=args, predColl=predColl, bgs=bgs)
model.train_model()
def define_preds(self):
nC=6
nD=16
C = ['%d'%i for i in range(nC)]
D = ['%d'%i for i in range(nD)]
self.Constants = dict( { 'C':C, 'D':D}) #, 'N':['%d'%i for i in range(6)] })
self.predColl = PredCollection (self.Constants)
self.predColl.add_pred(dname='pos' ,arguments=['C','D' ])
self.predColl.add_pred(dname='question' ,arguments=['C'])
self.predColl.add_pred(dname='eq' ,arguments=['D','D' ])
self.predColl.add_pred(dname='ltD' ,arguments=['D','D','D'])
self.predColl.add_pred(dname='gtD' ,arguments=['D','D','D'])
self.predColl.add_pred(dname='left' ,arguments=['D'])
self.predColl.add_pred(dname='button' ,arguments=['D'])
for i in range(self.q_dim):
self.predColl.add_pred(dname='is_q_%d'%i ,arguments=[])
for i in range(nC):
self.predColl.add_pred(dname='is_color_%d'%i ,arguments=['C'])
# ,pFunc =
# DNF('obj',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['is_l_0(A)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='or')
self.predColl.add_pred(dname='rectangle' ,arguments=['C'])
self.predColl.add_pred(dname='exist' ,arguments=['C'])
self.predColl.add_pred(dname='eqC' ,arguments=['C','C'],variables=['D','D'] ,pFunc =
DNF('eqC',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['pos(A,C), pos(B,D), eq(C,D)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='or')
self.predColl.add_pred(dname='closer',arguments=['C','C','C'], variables=['D','D','D'] ,pFunc =
DNF('closer',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['pos(A,D), pos(B,E), pos(C,F), exist(A), exist(B), exist(C), ltD(D,E,F)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='or')
self.predColl.add_pred(dname='farther',arguments=['C','C','C'], variables=['D','D','D'] ,pFunc =
DNF('farther',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['pos(A,D), pos(B,E), pos(C,F), exist(A), exist(B), exist(C), gtD(D,E,F)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='or')
self.predColl.add_pred(dname='closest',arguments=['C','C'], variables=['C'] ,pFunc =
DNF('closest',terms=4,init=[1,.1,-1,.1],sig=2,init_terms=['closer(A,C,B)','not exist(A)','not exist(B)','eqC(A,B)'],predColl=self.predColl,fast=True,neg=True) , use_neg=True, Fam='eq')
self.predColl.add_pred(dname='farthest',arguments=['C','C'], variables=['C'] ,pFunc =
DNF('farthest',terms=4,init=[1,.1,-1,.1],sig=2,init_terms=['farther(A,C,B)','not exist(A)','not exist(B)','eqC(A,B)'],predColl=self.predColl,fast=True,neg=True) , use_neg=True, Fam='eq')
exc = ['CL_%d'%i for i in range(self.a_dim)]
exc=[]
for k in range(0,self.a_dim):
# # # # exc = ['CL_%d'%i for i in range(k,self.a_dim)]
# # # # self.predColl.add_pred(dname='CL0_%d'%k,oname='CL_%d'%k,arguments=[] , variables=[] , pFunc = DNF('CL0_%d'%k,predColl=self.predColl,terms=10 ,init=[-1,.1,-1,.1],sig=2) ,use_neg=True, Fam='eq', exc_preds=exc )
# # self.predColl.add_pred(dname='CL1_%d'%k,oname='CL_%d'%k,arguments=[] , variables=['D'] , pFunc = DNF('CL1_%d'%k,predColl=self.predColl,terms=6 ,init=[-1,.1,-1,.1],sig=2) ,use_neg=True, Fam='eq',exc_conds=[('*','rep1') ] )
if k==0:
post_terms=[]
else:
post_terms=[ ('and', 'not CL_%d()'%j ) for j in range(k)]
post_terms=[]
self.predColl.add_pred(dname='CL_%d'%k,oname='CL_%d'%k,arguments=[] , variables=['C','C','D'] , pFunc = DNF('CL_%d'%k,predColl=self.predColl,terms=14,init=[-1,-1,-1,.1],sig=2, post_terms=post_terms) ,use_neg=True, Fam='eq',exc_conds=[('*','rep1') ] ,exc_preds=exc ) #
# # # # self.predColl.add_pred(dname='CL_%d'%k,oname='CL_%d'%k,arguments=[] , variables=['D','D'] , pFunc = MLP('CL_%d'%k,dims=[200,1], acts=[tf.nn.relu,tf.sigmoid] ) ,use_neg=False, Fam='eq', exc_preds=exc )
self.predColl.initialize_predicates()
self.bg = Background( self.predColl )
# self.bg.add_backgroud('notExist', ('%d'%(nD-1),))
for i in range(nC):
self.bg.add_backgroud('is_color_%d'%i, ('%d'%i,))
for i in range(nD):
ri,ci=int(i//4),int(i%4)
if ri>=2:
self.bg.add_backgroud('button', ('%d'%i,))
if ci<2:
self.bg.add_backgroud('left', ('%d'%i,))
# self.bg.add_backgroud('-%d'%i , ('%d'%i,) )
self.bg.add_backgroud('eq', ('%d'%i,'%d'%i))
for j in range(nD):
rj,cj=int(j//4),int(j%4)
for k in range(nD):
rk,ck=int(k//4),int(k%4)
d1=(ri-rj)**2+(ci-cj)**2
d2=(ri-rk)**2+(ci-ck)**2
if(d1<d2 and i!=j and i!=k and j!=k):
self.bg.add_backgroud('ltD', ('%d'%i,'%d'%j,'%d'%k))
if(d1>d2 and i!=j and i!=k and j!=k):
self.bg.add_backgroud('gtD', ('%d'%i,'%d'%j,'%d'%k))
a = '%d'%i
# self.bg.add_backgroud('is_c_%d'%i , (a,) )
# self.bg.add_backgroud('is_r_%d'%i , (a,) )
# self.bg.add_backgroud('eqC' , (a,a) )
# self.bg.add_backgroud('eqR' , (a,a) )
# for j in range(nC):
# if i<j:
# self.bg.add_backgroud('ltC', ('%d'%i, '%d'%j))
# self.bg.add_backgroud('ltR', ('%d'%i, '%d'%j))
bg_set=[]
self.X0=OrderedDict()
for p in self.predColl.outpreds:
if p.oname not in bg_set:
tmp = tf.expand_dims( tf.constant( self.bg.get_X0(p.oname) ,tf.float32) , 0)
self.X0[p.oname] = tf.tile( tmp , [self.batch_size,1] )
print('displaying config setting...')
# for arg in vars(args):
# print( '{}-{}'.format ( arg, getattr(args, arg) ) )
self.mdl = ILPRLEngine( args=self.args_ilp ,predColl=self.predColl ,bgs=None )
def define_preds(self):
X = ['%d' % i for i in range(self.DIM1)]
Y = ['%d' % i for i in range(self.DIM2)]
self.Constants = dict({'N': X, 'Y': Y})
self.predColl = PredCollection(self.Constants)
self.predColl.add_pred(dname='sameX', arguments=['N', 'N'])
self.predColl.add_pred(dname='sameY', arguments=['Y', 'Y'])
self.predColl.add_pred(dname='X_U', arguments=['N'])
self.predColl.add_pred(dname='X_D', arguments=['N'])
self.predColl.add_pred(dname='Y_L', arguments=['Y'])
self.predColl.add_pred(dname='Y_R', arguments=['Y'])
self.predColl.add_pred(dname='ltY', arguments=['Y', 'Y'])
self.predColl.add_pred(dname='close', arguments=['Y', 'Y'])
self.predColl.add_pred(dname='agent', arguments=['N', 'Y'])
self.predColl.add_pred(dname='predLR', arguments=['N', 'Y'])
self.predColl.add_pred(dname='predRL', arguments=['N', 'Y'])
self.predColl.add_pred(dname='food', arguments=['N', 'Y'])
count_type = 'max'
self.predColl.add_pred(dname='pred',
arguments=['N', 'Y'],
variables=[],
pFunc=DNF(
'pred',
init_terms=['predLR(A,B)', 'predRL(A,B)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type='or',
arg_funcs=[])
self.predColl.add_pred(dname='agentX',
arguments=['N'],
variables=['Y'],
pFunc=DNF('agentX',
init_terms=['agent(A,B)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type='or',
arg_funcs=[])
self.predColl.add_pred(dname='agentY',
arguments=['Y'],
variables=['N'],
pFunc=DNF('agentY',
init_terms=['agent(B,A)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type='or',
arg_funcs=[])
self.predColl.add_pred(
dname='C1',
arguments=[],
variables=['N', 'N'],
pFunc=DNF('C1',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=['agentX(A), agentX(B), not sameX(A,B)'],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq',
count_type='max',
arg_funcs=[])
self.predColl.add_pred(
dname='C2',
arguments=[],
variables=['Y', 'Y'],
pFunc=DNF('C2',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=['agentY(A), agentY(B), not close(A,B)'],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq',
count_type='max',
arg_funcs=[])
self.predColl.add_pred(dname='C3',
arguments=[],
variables=['N', 'Y'],
pFunc=DNF('C3',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=[
'predLR(A,B), predRL(A,B)',
'agent(A,B), predRL(A,B)',
'agent(A,B), predLR(A,B)',
],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq',
count_type='max',
arg_funcs=[])
self.predColl.add_pred(dname='C4',
arguments=[],
variables=['N', 'Y'],
pFunc=DNF('C4',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=['agent(A,B)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type='max',
arg_funcs=[])
excs = [
'action_noop', 'action_up', 'action_down', 'action_left',
'action_right', 'Q'
]
w = [-1, .1, -3, .1]
self.predColl.add_pred(dname='en_up',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF(
'en_up',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=[
'agent(A,B), X_U(A)',
'agent(A,B), pred(M_A,C), close(B,C)'
],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type=count_type,
count_th=100,
arg_funcs=['M'])
self.predColl.add_pred(dname='en_down',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF(
'en_down',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=[
'agent(A,B), X_D(A)',
'agent(A,B), pred(P_A,C), close(B,C)'
],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type=count_type,
count_th=100,
arg_funcs=['P'])
self.predColl.add_pred(
dname='en_right',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('en_right',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=[
'agent(A,B), Y_R(B)',
'agent(A,B), pred(A,C), close(B,C), ltY(B,C)'
],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type=count_type,
count_th=100,
arg_funcs=[])
self.predColl.add_pred(
dname='en_left',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('en_left',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
init_terms=[
'agent(A,B), Y_L(B)',
'agent(A,B), pred(A,C), close(B,C), ltY(C,B)'
],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type=count_type,
count_th=100,
arg_funcs=[])
self.predColl.add_pred(
dname='en_noop',
arguments=[],
variables=[],
pFunc=DNF(
'en_noop',
terms=1,
init=[-1, .1, -1, .1],
sig=2,
#init_terms=['agent(A,B), predRL(A,C), ltY(C,B)','agent(A,B), predLR(A,C), ltY(B,C)' ],
init_terms=['en_right()', 'en_left()'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq',
count_type=count_type,
count_th=100)
pt = [('and', 'not en_noop()')]
self.predColl.add_pred(dname='action_noop',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('action_noop',
terms=8,
init=w,
sig=2,
predColl=self.predColl,
fast=False,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=excs,
count_type=count_type,
arg_funcs=['M'])
pt = [('and', 'not en_up()')]
self.predColl.add_pred(dname='action_up',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('action_up',
terms=8,
init=w,
sig=2,
predColl=self.predColl,
fast=False,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=excs,
count_type=count_type,
count_th=100,
arg_funcs=['M'])
pt = [('and', 'not en_right()')]
self.predColl.add_pred(dname='action_right',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('action_right',
terms=8,
init=w,
sig=2,
predColl=self.predColl,
fast=False,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=excs,
count_type=count_type,
arg_funcs=['M'])
pt = [('and', 'not en_left()')]
self.predColl.add_pred(dname='action_left',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('action_left',
terms=8,
init=w,
sig=2,
predColl=self.predColl,
fast=False,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=excs,
count_type=count_type,
arg_funcs=['M'])
pt = [('and', 'not en_down()')]
self.predColl.add_pred(dname='action_down',
arguments=[],
variables=['N', 'Y', 'Y'],
pFunc=DNF('action_down',
terms=8,
init=w,
sig=2,
predColl=self.predColl,
fast=False,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=excs,
count_type=count_type,
arg_funcs=['M'])
self.predColl.initialize_predicates()
self.bg = Background(self.predColl)
self.bg.add_backgroud('X_U', ('%d' % 0, ))
self.bg.add_backgroud('X_D', ('%d' % (self.DIM1 - 1), ))
self.bg.add_backgroud('Y_L', ('%d' % 0, ))
self.bg.add_backgroud('Y_R', ('%d' % (self.DIM2 - 1), ))
for i in range(self.DIM1):
self.bg.add_backgroud('sameX', (
'%d' % i,
'%d' % i,
))
for i in range(self.DIM2):
self.bg.add_backgroud('sameY', (
'%d' % i,
'%d' % i,
))
for i in range(self.DIM2):
for j in range(self.DIM2):
if i < j:
self.bg.add_backgroud('ltY', (
'%d' % i,
'%d' % j,
))
if abs(i - j) < 2:
self.bg.add_backgroud('close', (
'%d' % i,
'%d' % j,
))
print('displaying config setting...')
self.mdl = ILPRLEngine(args=self.args,
predColl=self.predColl,
bgs=None)
def define_preds(self):
Box = ['%d' % i for i in range(self.num_box + 1)]
Ds = ['%d' % i for i in range(self.num_box + 1)]
self.Constants = dict({'C': Box, 'D': Ds})
self.predColl = PredCollection(self.Constants)
self.predColl.add_pred(dname='posH', arguments=['C', 'D'])
self.predColl.add_pred(dname='posV', arguments=['C', 'D'])
self.predColl.add_pred(dname='is_one', arguments=['D'])
self.predColl.add_pred(dname='lt', arguments=['D', 'D'])
self.predColl.add_pred(dname='inc', arguments=['D', 'D'])
self.predColl.add_pred(dname='same', arguments=['C', 'C'])
self.predColl.add_pred(dname='is_floor', arguments=['C'])
self.predColl.add_pred(dname='is_blue', arguments=['C'])
self.predColl.add_pred(dname='same_col',
arguments=['C', 'C'],
variables=['D'],
pFunc=DNF('same_col',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['posH(A,C), posH(B,C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(
dname='above',
arguments=['C', 'C'],
variables=['D', 'D'],
pFunc=DNF(
'above',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['same_col(A,B), posV(A,C), posV(B,D), lt(D,C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(
dname='below',
arguments=['C', 'C'],
variables=['D', 'D'],
pFunc=DNF(
'below',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['same_col(A,B), posV(A,C), posV(B,D), lt(C,D)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(
dname='on',
arguments=['C', 'C'],
variables=['D', 'D'],
pFunc=DNF('on',
terms=2,
init=[1, .1, -1, .1],
sig=2,
init_terms=[
'is_floor(B), posV(A,C), is_one(C)',
'same_col(A,B), posV(A,C), posV(B,D), inc(D,C)'
],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(dname='isCovered',
arguments=['C'],
variables=['C'],
pFunc=DNF(
'isCovered',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['on(B,A), not is_floor(A)'],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq')
self.predColl.add_pred(
dname='lower',
oname='lower',
arguments=['C', 'C'],
variables=['D', 'D'],
pFunc=DNF('lower',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['posV(A,C), posV(B,D), lt(C,D)'],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq')
self.predColl.add_pred(
dname='moveable',
oname='moveable',
arguments=['C', 'C'],
variables=[],
pFunc=DNF(
'moveable',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=[
'not isCovered(A), not isCovered(B), not same(A,B), not is_floor(A), not on(A,B), not is_blue(A), not is_floor(B), not lower(B,A)'
],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq')
pt = [
('and', 'moveable(A,B)'),
]
self.predColl.add_pred(dname='move',
oname='move',
arguments=['C', 'C'],
variables=[],
pFunc=DNF('move',
terms=4,
init=[1, -1, -1, .1],
sig=2,
predColl=self.predColl,
fast=False,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=[],
exc_conds=[('*', 'rep1')])
self.predColl.initialize_predicates()
self.bg = Background(self.predColl)
#define backgrounds
self.bg.add_backgroud('is_floor', ('0', ))
self.bg.add_backgroud('is_one', ('1', ))
self.bg.add_backgroud('is_blue', ('1', ))
for i in range(self.num_box + 1):
self.bg.add_backgroud('same', ('%d' % i, '%d' % i))
if '%d' % (i + 1) in Ds:
self.bg.add_backgroud('inc', ('%d' % i, '%d' % (i + 1)))
for i in range(self.num_box + 1):
for j in range(self.num_box + 1):
if i < j:
self.bg.add_backgroud('lt', ('%d' % i, '%d' % (j)))
print('displaying config setting...')
self.mdl = ILPRLEngine(args=self.args,
predColl=self.predColl,
bgs=None)
def define_preds(self):
nCOLOR = 10
Colors = [i for i in range(nCOLOR)]
Pos = [i for i in range(12)]
# Ds = ['%d'%i for i in range(self.num_box)]
self.Constants = dict({
'C': Colors,
'P': Pos,
'Q': Pos
}) #, 'N':['%d'%i for i in range(6)] })
self.predColl = PredCollection(self.Constants)
self.predColl.add_pred(dname='color', arguments=['P', 'Q', 'C'])
for i in range(nCOLOR):
self.predColl.add_pred(dname='is%d' % i, arguments=['C'])
# self.predColl.add_pred(dname='Pos0' ,arguments=['P'])
# self.predColl.add_pred(dname='Pos11' ,arguments=['P'])
self.predColl.add_pred(dname='has_key', arguments=['C'])
self.predColl.add_pred(dname='neq', arguments=['C', 'C'])
# self.predColl.add_pred(dname='lt' ,arguments=['P','P'])
# self.predColl.add_pred(dname='incp' ,arguments=['P','P'])
self.predColl.add_pred(dname='incq', arguments=['Q', 'Q'])
self.predColl.add_pred(dname='isBK',
arguments=['P', 'Q'],
variables=['C'],
pFunc=DNF('isBK',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['color(A,B,C), is0(C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(dname='isAgent',
arguments=['P', 'Q'],
variables=['C'],
pFunc=DNF('isAgent',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['color(A,B,C), is1(C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(dname='isGem',
arguments=['P', 'Q'],
variables=['C'],
pFunc=DNF('isGem',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['color(A,B,C), is2(C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(
dname='isItem',
arguments=['P', 'Q'],
variables=[],
pFunc=DNF('isItem',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['not isBK(A,B), not isAgent(A,B)'],
predColl=self.predColl,
fast=True),
use_neg=True,
Fam='eq')
# self.predColl.add_pred(dname='sameColor' ,arguments=['P','P','P','P'], variables=['C'] ,pFunc =
# DNF('sameColor',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['color(A,B,E), color(C,D,E)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='or')
self.predColl.add_pred(
dname='locked',
arguments=['P', 'Q'],
variables=['Q'],
pFunc=DNF('locked',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['isItem(A,B), isItem(A,C), incq(B,C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
self.predColl.add_pred(
dname='isLock',
arguments=['P', 'Q'],
variables=['Q'],
pFunc=DNF('isLock',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['isItem(A,B), isItem(A,C), incq(C,B)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='eq')
# self.predColl.add_pred(dname='locked1' ,arguments=['P','P'], variables=['P'] ,pFunc =
# DNF('locked1',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['isItem(A,B), isItem(C,B), inc(C,B)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='or')
#self.predColl.add_pred(dname='LockColor' ,arguments=['C','C'], variables=['P','Q','Q'] ,pFunc =
# DNF('LockColor',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['isItem(C,D), isItem(C,E), color(C,D,A), color(C,E,B), incq(D,E)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='eq')
#self.predColl.add_pred(dname='loosekey' ,arguments=['P','Q'], variables=[] ,pFunc =
# DNF('loosekey',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['isItem(A,B), not isLock(A,B), not locked(A,B)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='eq')
#self.predColl.add_pred(dname='key_color' ,arguments=['P','Q','C'], variables=['Q'] ,pFunc =
## DNF('key_color',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['isItem(A,B), isLock(A,B), color(A,B,C)',
# 'isItem(A,B), locked(A,D), incq(D,B), color(A,D,C)' ],predColl=self.predColl,fast=True) , use_neg=True, Fam='eq')
#self.predColl.add_pred(dname='inGoal' ,arguments=['C'], variables=['C'] ,pFunc =
# DNF('inGoal',terms=2,init=[1,.1,-1,.1],sig=2,init_terms=['is2(A)' , 'inGoal(B), LockColor(B,A)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='or')
# self.predColl.add_pred(dname='valid',oname='valid',arguments=['P','P'] , variables=['C'] ,pFunc =
# DNF('valid',terms=2,init_terms=['loosekey(A,B)', 'isItem(A,B), isLock(A,B), color(A,B,C), has_key(C)'],sig=2,predColl=self.predColl,fast=True ) , use_neg=True, Fam='eq')
#,post_terms=[('and','isItem(A,B)')] )
# self.predColl.add_pred(dname='move',oname='move',arguments=['P','P'] , variables=['C'] ,pFunc =
# DNF('move',terms=4,init=[-1,.1,-1,.1],sig=2,predColl=self.predColl,fast=False,post_terms=[('and','not isBK(A,B)')]) , use_neg=True, Fam='eq')
# self.predColl.add_pred(dname='move',oname='move',arguments=['P','P'] , variables=['C','C'] ,pFunc =
# DNF('move',terms=8,init=[1,-1,-2,.3],sig=1,predColl=self.predColl,fast=False,post_terms=[('and','isItem(A,B)')]) , use_neg=True, Fam='eq', exc_preds=['move']+[ 'is%d'%i for i in range(nCOLOR)],exc_conds=[('*','rep1') ])
# DNF('move',terms=5,init=[1,-3,-1,.1],sig=2,predColl=self.predColl,fast=False,post_terms=[('and','isItem(A,B)')]) , use_neg=False, Fam='eq', exc_preds=['move']+[ 'is%d'%i for i in range(nCOLOR)],exc_conds=[('*','rep1') ])
# DNF('move',terms=10,init=[1,-1,-1,.1],sig=2,predColl=self.predColl,fast=False,post_terms=[('and','isItem(A,B)'),('or','loosekey(A,B)')]) , use_neg=True, Fam='eq', exc_preds=['move', 'color'])
# DNF('move',terms=5,init_terms=['color(A,B,C), inGoal(C), loosekey(A,B)','key_color(A,B,D), color(A,B,C), inGoal(D), isLock(A,B), inGoal(D), LockColor(D,C), has_key(C)'],sig=2,predColl=self.predColl,fast=True ) , use_neg=True, Fam='eq')
# DNF('move',terms=2,init_terms=['loosekey(A,B)', 'isLock(A,B), color(A,B,C), has_key(C)'],sig=2,predColl=self.predColl,fast=True ) , use_neg=True, Fam='eq')
# for i in range(10):
# self.predColl.add_pred(dname='aux%d'%i,oname='aux%d'%i,arguments=['P','Q'] , variables=['C','C'] ,pFunc =
# CONJ('aux%d'%i,init=[-1,1],init_terms=[],sig=1,predColl=self.predColl,post_terms=[]) , use_neg=False, Fam='eq', exc_preds=['move']+[ '%d'%i for i in range(nCOLOR)]+['aux%d'%i for i in range(10)],exc_conds=[('*','rep1') ])
# it=[]
#self.predColl.add_pred(dname='Q',oname='Q',arguments=['P','Q'] , variables=['P', 'C' ] ,pFunc =
# DNF('Q',terms=6,init=[-1,.1,-1,1],init_terms=[],sig=1,predColl=self.predColl ) , use_neg=True, Fam='eq', exc_preds=['move','Q']+[ 'is%d'%i for i in range(nCOLOR)],exc_conds=[('*','rep1') ])
#self.predColl.add_pred(dname='move',oname='move',arguments=['P','Q'] , variables=['C' ] ,pFunc =
# DNF('move',terms=6,init=[1,-1,-2,1],init_terms=[],sig=2,predColl=self.predColl,post_terms=[('and','isItem(A,B)'),('and','inGoal(C)') ]) , use_neg=True, Fam='eq', exc_preds=['move','Q']+[ 'is%d'%i for i in range(nCOLOR)],exc_conds=[('*','rep1') ])
pt = [('and', 'isItem(A,B)')]
pt = []
self.predColl.add_pred(dname='move',
oname='move',
arguments=['P', 'Q'],
variables=['C'],
pFunc=DNF('move',
terms=6,
init=[-1, .1, -1, .1],
init_terms=[],
sig=1,
predColl=self.predColl,
post_terms=pt),
use_neg=True,
Fam='eq',
exc_preds=['move', 'Q'] +
['is%d' % i for i in range(nCOLOR)],
exc_conds=[('*', 'rep1')])
# for i in range(5):
# if i==0:
# tp="eq"
# # it=['color(A,B,C), inGoal(C), loosekey(A,B)']
# else:
# tp="or"
# it=[]
# vvv=1
# vv=-1
# Vars = ['C','C' ]
# if i==5:
# Vars=['C','C']
# vvv=2
# vv=-1
# # if i==1:
# # it=['key_color(A,B,D), color(A,B,C), isLock(A,B), inGoal(D), LockColor(D,C), has_key(C)']
# self.predColl.add_pred(dname='move%d'%i,oname='move',arguments=['P','Q'] , variables=Vars ,pFunc =
# CONJ('move%d'%i,init=[vv,vvv],init_terms=it,sig=1,predColl=self.predColl,post_terms=[('and','isItem(A,B)')]) , use_neg=False, Fam=tp, exc_preds=['move']+[ '%d'%i for i in range(nCOLOR)],exc_conds=[('*','rep1') ])
# self.predColl.add_pred(dname='move2',oname='move',arguments=['P','P'] , variables=['C'] ,pFunc =
# CONJ('move2',init=[-1,.1],sig=2,predColl=self.predColl,fast=False) , use_neg=True, Fam='eq', exc_preds=['move'],exc_conds=[('*','rep1') ])
# MLP('move',dims=[64,64,1], acts=[relu1,relu1,tf.nn.sigmoid] ) , use_neg=False, Fam='eq')
# DNF('move',terms=10,init=[1,-1,-1,.1],sig=2,predColl=self.predColl,fast=False,post_terms=[('and','isItem(A,B)'),('or','loosekey(A,B)')]) , use_neg=True, Fam='eq', exc_preds=['move', 'color'])
# self.predColl.add_pred(dname='move',oname='move',arguments=['P','P'] , variables=['C'] ,pFunc =
# DNF('move',terms=2,init_terms=['loosekey(A,B)', 'isLock(A,B), color(A,B,C), has_key(C)'],sig=2,predColl=self.predColl,fast=True ) , use_neg=True, Fam='eq')
# self.predColl.add_pred(dname='has_key' ,arguments=['C'], variables=['P','P'] ,pFunc =
# DNF('has_key',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['loosekey(B,C), move(B,C), color(B,C,A)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='or')
# self.predColl.add_pred(dname='enableLeft' ,arguments=['P','P'], variables=['P'] ,pFunc =
# DNF('enableLeft',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['agent(A,B), inc(C,A), not pos0(C)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='or')
# self.predColl.add_pred(dname='enableRight' ,arguments=['P','P'], variables=['P'] ,pFunc =
# DNF('enableRight',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['agent(A,B), not locked(A,B), not Pos0(A)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='or')
# self.predColl.add_pred(dname='enableLeft' ,arguments=['P','P'], variables=['P'] ,pFunc =
# DNF('enableLeft',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['agent(A,B), not locked(A,B), not Pos0(A)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='or')
# self.predColl.add_pred(dname='enableLeft' ,arguments=['P','P'], variables=['P'] ,pFunc =
# DNF('enableLeft',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['agent(A,B), not locked(A,B), not Pos0(A)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='or')
# self.predColl.add_pred(dname='moveLeft',oname='moveLeft',arguments=[] , variables=['P','P','C'] ,pFunc =
# DNF('moveLeft',terms=6,init=[1,.1,-1,.1],sig=2,predColl=self.predColl,fast=False) , use_neg=True, Fam='eq')
# self.predColl.add_pred(dname='moveRight',oname='moveRight',arguments=[] , variables=['P','P','C'] ,pFunc =
# DNF('moveRight',terms=6,init=[1,.1,-1,.1],sig=2,predColl=self.predColl,fast=False) , use_neg=True, Fam='eq')
# self.predColl.add_pred(dname='moveLeft',oname='moveLeft',arguments=[] , variables=['P','P','C'] ,pFunc =
# DNF('moveLeft',terms=6,init=[1,.1,-1,.1],sig=2,predColl=self.predColl,fast=False) , use_neg=True, Fam='eq')
# self.predColl.add_pred(dname='moveLeft',oname='moveLeft',arguments=[] , variables=['P','P','C'] ,pFunc =
# DNF('moveLeft',terms=6,init=[1,.1,-1,.1],sig=2,predColl=self.predColl,fast=False) , use_neg=True, Fam='eq')
self.predColl.initialize_predicates()
self.bg = Background(self.predColl)
#define backgrounds
self.bg.add_backgroud('Pos0', (0, ))
self.bg.add_backgroud('Pos11', (11, ))
for i in range(nCOLOR):
self.bg.add_backgroud('is%d' % i, (i, ))
for j in range(nCOLOR):
if i != j:
self.bg.add_backgroud('neq', (i, i))
for i in range(12):
self.bg.add_backgroud('eq', (i, i))
for j in range(12):
if i < j:
self.bg.add_backgroud('lt', (i, j))
if j == i + 1:
# self.bg.add_backgroud('incp',(i,j) )
self.bg.add_backgroud('incq', (i, j))
print('displaying config setting...')
self.mdl = ILPRLEngine(args=self.args,
predColl=self.predColl,
bgs=None)
def define_preds(self):
nCOLOR = 10
Colors=[i for i in range(nCOLOR)]
Pos = [i for i in range(12)]
# Ds = ['%d'%i for i in range(self.num_box)]
self.Constants = dict( {'C':Colors,'P':Pos,'Q':Pos}) #, 'N':['%d'%i for i in range(6)] })
self.predColl = PredCollection (self.Constants)
self.predColl.add_pred(dname='color' ,arguments=['P','Q','C'])
for i in range(nCOLOR):
self.predColl.add_pred(dname='is%d'%i ,arguments=['C'])
self.predColl.add_pred(dname='has_key' ,arguments=['C'] )
self.predColl.add_pred(dname='neq' ,arguments=['C','C'])
self.predColl.add_pred(dname='incq' ,arguments=['Q','Q'])
self.predColl.add_pred(dname='isBK' ,arguments=['P','Q'], variables=['C'] ,pFunc =
DNF('isBK',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['color(A,B,C), is0(C)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='eq')
self.predColl.add_pred(dname='isAgent' ,arguments=['P','Q'], variables=['C'] ,pFunc =
DNF('isAgent',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['color(A,B,C), is1(C)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='eq')
self.predColl.add_pred(dname='isGem' ,arguments=['P','Q'], variables=['C'] ,pFunc =
DNF('isGem',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['color(A,B,C), is2(C)'],predColl=self.predColl,fast=True) , use_neg=False, Fam='eq')
self.predColl.add_pred(dname='isItem' ,arguments=['P','Q'], variables=[] ,pFunc =
DNF('isItem',terms=1,init=[1,.1,-1,.1],sig=2,init_terms=['not isBK(A,B), not isAgent(A,B)'],predColl=self.predColl,fast=True) , use_neg=True, Fam='eq')
pt=[]
self.predColl.add_pred(dname='move',oname='move',arguments=['P','Q'] , variables=['C' ] ,pFunc =
DNF('move',terms=6,init=[-1,.1,-1,.1],init_terms=[],sig=1,predColl=self.predColl,post_terms=pt) , use_neg=True, Fam='eq', exc_preds=['move','Q']+[ 'is%d'%i for i in range(nCOLOR)],exc_conds=[('*','rep1') ])
self.predColl.initialize_predicates()
self.bg = Background( self.predColl )
#define backgrounds
self.bg.add_backgroud('Pos0',(0,) )
self.bg.add_backgroud('Pos11',(11,) )
for i in range(nCOLOR):
self.bg.add_backgroud('is%d'%i , (i,))
for j in range(nCOLOR):
if i!=j:
self.bg.add_backgroud('neq',(i,i) )
for i in range(12):
self.bg.add_backgroud('eq',(i,i) )
for j in range(12):
if i<j:
self.bg.add_backgroud('lt',(i,j) )
if j==i+1:
# self.bg.add_backgroud('incp',(i,j) )
self.bg.add_backgroud('incq',(i,j) )
print('displaying config setting...')
self.mdl = ILPRLEngine( args=self.args ,predColl=self.predColl ,bgs=None )
def define_preds(self):
X = ['%d'%i for i in range(self.DIM1)]
Y = ['%d'%i for i in range(self.DIM2)]
self.Constants = dict( {'N':X,'Y':Y})
self.predColl = PredCollection (self.Constants)
# self.predColl.add_pred(dname='agentX' ,arguments=['N'])
# self.predColl.add_pred(dname='agentY' ,arguments=['Y'])
for i in range(self.DIM1):
if i==0 or i==self.DIM1-1:
self.predColl.add_pred(dname='X_%d'%i,arguments=['N'])
for i in range(self.DIM2):
if i==0 or i==self.DIM2-1:
self.predColl.add_pred(dname='Y_%d'%i,arguments=['Y'])
# self.predColl.add_pred(dname='f_1',arguments=['N','Y'], variables=[ ],pFunc =
# DNF('f_1',init_terms=['agentX(A), agentY(B)' ],predColl=self.predColl,fast=True) , use_neg=False, Fam='eq')
self.predColl.add_pred(dname='f_1',arguments=['N','Y'])
self.predColl.add_pred(dname='f_2',arguments=['N','Y'])
self.predColl.add_pred(dname='f_3',arguments=['N','Y'])
self.predColl.add_pred(dname='f_4',arguments=['N','Y'])
self.predColl.add_pred(dname='ltY',arguments=['Y','Y'])
self.predColl.add_pred(dname='close',arguments=['Y','Y'])
count_type = 'max'
excs = ['action_noop', 'action_up', 'action_down','action_left','action_right', 'Q']
w = [-1,.1,-2,.1]
Alicount_type = None
self.predColl.add_pred(dname='en_up' ,arguments=[], variables=[ 'N','Y', 'Y'],pFunc =
DNF('en_up',terms=1,init=[-1,.1,-1,.1],sig=2,
init_terms=['f_1(A,B), X_0(A)','f_1(A,B), f_3(M_A,C), close(B,C), ltY(B,C)', 'f_1(A,B), f_2(M_A,C), close(B,C), ltY(C,B)' ],
predColl=self.predColl,fast=True) , use_neg=False, Fam='eq',count_type=count_type, count_th=100,arg_funcs=['M'])
self.predColl.add_pred(dname='en_down' ,arguments=[ ], variables=['N','Y', 'Y'],pFunc =
DNF('en_down',terms=1,init=[-1,.1,-1,.1],sig=2,
init_terms=['f_1(A,B), X_%d(A)'%(self.DIM1-1) , 'f_1(A,B), f_3(P_A,C), close(B,C), ltY(B,C)', 'f_1(A,B), f_2(P_A,C), close(B,C), ltY(C,B)' ]
,predColl=self.predColl,fast=True) , use_neg=False, Fam='eq',count_type=count_type, count_th=100,arg_funcs=['P'])
self.predColl.add_pred(dname='en_right' ,arguments=[ ], variables=['N','Y','Y'],pFunc =
DNF('en_right',terms=1,init=[-1,.1,-1,.1],sig=2,
init_terms=['f_1(A,B), Y_%d(B)'%(self.DIM2-1) ,'f_1(A,B), f_3(A,C), close(B,C), ltY(B,C)' ,'f_1(A,B), f_2(A,C), close(B,C), ltY(B,C)' ],
predColl=self.predColl,fast=True) , use_neg=False, Fam='eq',count_type=count_type, count_th=100,arg_funcs=[ ])
self.predColl.add_pred(dname='en_left' ,arguments=[ ], variables=['N','Y','Y'],pFunc =
DNF('en_left',terms=1,init=[-1,.1,-1,.1],sig=2,
init_terms=['f_1(A,B), Y_%d(B)'%0, 'f_1(A,B), f_2(A,C), close(B,C), ltY(C,B)','f_1(A,B), f_3(A,C), close(B,C), ltY(C,B)' ],
predColl=self.predColl,fast=True) , use_neg=False, Fam='eq',count_type=count_type, count_th=100,arg_funcs=[ ])
self.predColl.add_pred(dname='en_noop' ,arguments=[ ], variables=[],pFunc =
DNF('en_noop',terms=1,init=[-1,.1,-1,.1],sig=2,
#init_terms=['f_1(A,B), f_3(A,C), ltY(C,B)','f_1(A,B), f_2(A,C), ltY(B,C)' ],
init_terms=['en_right()','en_left()' ],
predColl=self.predColl,fast=True) , use_neg=False, Fam='eq',count_type=count_type, count_th=100)
pt = [('and', 'not en_noop()') ]
self.predColl.add_pred(dname='action_noop',arguments=[] , variables=['N','Y','Y' ] ,pFunc =
DNF('action_noop',terms=8,init=w,sig=2,predColl=self.predColl,fast=False,post_terms=pt) , use_neg=True, Fam='eq' ,
exc_preds=excs,count_type=count_type,arg_funcs=['M'])
pt = [('and', 'not en_up()') ]
self.predColl.add_pred(dname='action_up',arguments=[] , variables=['N','Y','Y' ] ,pFunc =
DNF('action_up',terms=8,init=w,sig=2,predColl=self.predColl,fast=False,post_terms=pt) , use_neg=True, Fam='eq' ,
exc_preds=excs,count_type=count_type, count_th=100,arg_funcs=[ 'M'])
pt = [('and', 'not en_right()') ]
self.predColl.add_pred(dname='action_right',arguments=[] , variables=['N','Y','Y' ] ,pFunc =
DNF('action_right',terms=8,init=w,sig=2,predColl=self.predColl,fast=False,post_terms=pt) , use_neg=True, Fam='eq' ,
exc_preds=excs,count_type=count_type,arg_funcs=['M'])
pt = [('and', 'not en_left()') ]
self.predColl.add_pred(dname='action_left',arguments=[] , variables=['N','Y', 'Y' ] ,pFunc =
DNF('action_left',terms=8,init=w,sig=2,predColl=self.predColl,fast=False,post_terms=pt) , use_neg=True, Fam='eq' ,
exc_preds=excs,count_type=count_type, arg_funcs=['M'])
pt = [('and', 'not en_down()') ]
self.predColl.add_pred(dname='action_down',arguments=[] , variables=['N','Y','Y' ] ,pFunc =
DNF('action_down',terms=8,init=w,sig=2,predColl=self.predColl,fast=False,post_terms=pt) , use_neg=True, Fam='eq' ,
exc_preds=excs,count_type=count_type, arg_funcs=['M'])
self.predColl.initialize_predicates()
self.bg = Background( self.predColl )
#define backgrounds
self.bg.add_backgroud('X_%d'%0 ,('%d'%0,) )
self.bg.add_backgroud('X_%d'%(self.DIM1-1) ,('%d'%(self.DIM1-1),) )
self.bg.add_backgroud('Y_%d'%0 ,('%d'%0,) )
self.bg.add_backgroud('Y_%d'%(self.DIM2-1) ,('%d'%(self.DIM2-1),) )
self.bg.add_backgroud('Y_%d'%0 ,('%d'%1,) )
self.bg.add_backgroud('Y_%d'%(self.DIM2-1) ,('%d'%(self.DIM2-2),) )
for i in range(self.DIM2):
for j in range(self.DIM2):
if i<=j+1:
self.bg.add_backgroud('ltY' ,('%d'%i,'%d'%j,) )
if abs(i-j)<3:
self.bg.add_backgroud('close' ,('%d'%i,'%d'%j,) )
print('displaying config setting...')
self.mdl = ILPRLEngine( args=self.args ,predColl=self.predColl ,bgs=None )
def define_preds(self):
nL = 6
nD = 16
D = ['%d' % i for i in range(nD)]
self.Constants = dict({'D': D})
self.predColl = PredCollection(self.Constants)
self.predColl.add_pred(dname='eq', arguments=['D', 'D'])
self.predColl.add_pred(dname='ltD', arguments=['D', 'D', 'D'])
self.predColl.add_pred(dname='gtD', arguments=['D', 'D', 'D'])
self.predColl.add_pred(dname='left', arguments=['D'])
self.predColl.add_pred(dname='button', arguments=['D'])
# self.predColl.add_pred(dname='top' ,arguments=['D'])
# self.predColl.add_pred(dname='right' ,arguments=['D'])
self.predColl.add_pred(dname='obj', arguments=['D'])
self.predColl.add_pred(dname='rectangle', arguments=['D'])
#instead of color(D,C) we define a set of is_
for i in range(nL):
self.predColl.add_pred(dname='is_color_%d' % i, arguments=['D'])
# for i in range(nD):
# self.predColl.add_pred(dname='is_d_%d'%i ,arguments=['D'])
for i in range(self.q_dim):
self.predColl.add_pred(dname='is_q_%d' % i, arguments=[])
self.predColl.add_pred(
dname='closer',
arguments=['D', 'D', 'D'],
variables=[],
pFunc=DNF('closer',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['obj(A), obj(B), obj(C), ltD(A,B,C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='or')
self.predColl.add_pred(
dname='farther',
arguments=['D', 'D', 'D'],
variables=[],
pFunc=DNF('farther',
terms=1,
init=[1, .1, -1, .1],
sig=2,
init_terms=['obj(A), obj(B), obj(C), gtD(A,B,C)'],
predColl=self.predColl,
fast=True),
use_neg=False,
Fam='or')
self.predColl.add_pred(dname='notClosest',
arguments=['D', 'D'],
variables=['D'],
pFunc=DNF('notClosest',
terms=3,
init=[1, .1, -1, .1],
sig=2,
init_terms=[
'closer(A,C,B)', 'not obj(A)',
'not obj(B)', 'eq(A,B)'
],
predColl=self.predColl,
fast=True,
neg=True),
use_neg=True,
Fam='eq')
self.predColl.add_pred(dname='notFarthest',
arguments=['D', 'D'],
variables=['D'],
pFunc=DNF('notFarthest',
terms=3,
init=[1, .1, -1, .1],
sig=2,
init_terms=[
'farther(A,C,B)', 'not obj(A)',
'not obj(B)', 'eq(A,B)'
],
predColl=self.predColl,
fast=True,
neg=True),
use_neg=True,
Fam='eq')
exc = ['CL_%d' % i for i in range(self.a_dim)]
self.predColl.add_pred(
dname='qa',
oname='qa',
arguments=['D'],
variables=[],
pFunc=DNF('qa',
predColl=self.predColl,
init_terms=[
'is_q_0(), is_color_0(A)', 'is_q_1(), is_color_1(A)',
'is_q_2(), is_color_2(A)', 'is_q_3(), is_color_3(A)',
'is_q_4(), is_color_4(A)', 'is_q_5(), is_color_5(A)'
],
fast=True),
use_neg=True,
Fam='eq',
exc_conds=[('*', 'rep1')],
exc_preds=exc)
for k in range(0, 10):
post_term = [('and', 'qa(A)')]
if k == 6:
post_term.append(('and', 'not rectangle(A)'))
if k == 7:
post_term.append(('and', 'rectangle(A)'))
post_terms = []
self.predColl.add_pred(dname='CL_%d' % k,
oname='CL_%d' % k,
arguments=[],
variables=['D', 'D'],
pFunc=DNF('CL_%d' % k,
terms=12,
init=[-1, -1, -1, -1],
sig=1,
predColl=self.predColl,
post_terms=post_term),
use_neg=True,
Fam='eq',
exc_preds=exc + ['eq', 'ltD', 'gtD'])
self.predColl.initialize_predicates()
self.bg = Background(self.predColl)
for i in range(nD):
self.bg.add_backgroud('eq', ('%d' % i, '%d' % i))
ri, ci = int(i // 4), int(i % 4)
Y1 = (ri + 0.5) * 28 + ri**2
X1 = (ci + 0.5) * 28 + ci**2
if Y1 > 64:
self.bg.add_backgroud('button', ('%d' % i, ))
if X1 < 64:
self.bg.add_backgroud('left', ('%d' % i, ))
for j in range(nD):
rj, cj = int(j // 4), int(j % 4)
for k in range(nD):
rk, ck = int(k // 4), int(k % 4)
Y2 = (rj + 0.5) * 28 + rj**2
X2 = (cj + 0.5) * 28 + cj**2
Y3 = (rk + 0.5) * 28 + rk**2
X3 = (ck + 0.5) * 28 + ck**2
d1 = 1.1 * (X1 - X2)**2 + (Y1 - Y2)**2
d2 = 1.1 * (X1 - X3)**2 + (Y1 - Y3)**2
if (d1 < d2 and i != j and i != k and j != k):
self.bg.add_backgroud('ltD',
('%d' % i, '%d' % j, '%d' % k))
if (d1 > d2 and i != j and i != k and j != k):
self.bg.add_backgroud('gtD',
('%d' % i, '%d' % j, '%d' % k))
bg_set = []
self.X0 = OrderedDict()
for p in self.predColl.outpreds:
if p.oname not in bg_set:
tmp = tf.expand_dims(
tf.constant(self.bg.get_X0(p.oname), tf.float32), 0)
self.X0[p.oname] = tf.tile(tmp, [self.batch_size, 1])
print('displaying config setting...')
self.mdl = ILPRLEngine(args=self.args_ilp,
predColl=self.predColl,
bgs=None)
