Added prediction and visualizations of prediction by the learned graph