set_display_label function

Show the probability of a specified label when visualizing a learner