graph looks better, compute loglikelihood on test set, network now also directly depends on robot configuration