Narine:

Captum · Model Interpretability for PyTorch


@sushmit_roy, if you want to do something  similar to scikitlearn, you can leave target=None and set forward_func as a scoring function.
forward_func can be something like
def my_forward_func(inputs, labels):
   return score(model(input), labels) # here score should be returned in form of a scalar tensor.

labels can be passed as additional_forward_args to the attribute method. Let me know if it works for you.