Describe the bug
If the batchsize is not -1 (i.e. not fullbatch), then the edge weights of the result graph obtained by calling sam.predict() are outside of the domain [0, 1]. This is consistent throughout all cdt-package-, python- and library versions and hardware configurations.
Steps to reproduce: Just set the parameter batch_size to something greater than 1 (eg. 2), test_epochs to sth. greater than 0 and call sam.predict().