Visualize the training data