Hello,
Currently I have a model to predict a binary target using NLLLoss() with the final layer going through a log_softmax() which gives me probability for target being 0 and probability for target being 1.  I would like to output 1 final probability in order to use the Captum package to calculate feature importance.  What is the correct combination of loss function and final activation function to give me one probability prediction instead of two?