I am working to use the code provided in this tutorial: Posit AI Blog: Audio classification with torch. The tutorial includes the following line to evaluate performance on the test dataset:
evaluate(fitted, test_dl)
However, I would like to return both the class labels and associated probabilities for the 'test_dl' dataset. How can I modify the code to do this? Thanks in advance!