Document set_probabilities function