Expanded transition probabilities visualization to include learned graph