{10,51} atom_embed(A) :- a_feats(A). [activation=identity]
atom_embed/1 [activation=identity]

{10,10} l1_embed(X) :- atom_embed(Y), bond(X,Y,B), <-1>. [aggregation=sum, activation=identity]
l1_embed/1 [activation=relu]
{10,10} l2_embed(X) :- l1_embed(Y), bond(X,Y,B), <-1>. [aggregation=sum, activation=identity]
l2_embed/1 [activation=identity]

{1,10} predict :- l2_embed(X). [aggregation=avg, activation=identity]
predict/0 [activation=sigmoid]