How can I modify the code in 'audio classification with torch' tutorial to return associated probabilities for each class?

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!

This topic was automatically closed 21 days after the last reply. New replies are no longer allowed.

If you have a query related to it or one of the replies, start a new topic and refer back with a link.