Following the `classification_sample`, implement a way to return the top N results from an inference along with the probabilities.