def __init__(self, predictor: Predictor):
CaptumAttribution.__init__(self, 'intgrad', predictor)
self.submodel = self.predictor._model.captum_sub_model()
IntegratedGradients.__init__(self, self.submodel)
def __init__(self, trainer, n_steps=20, internal_batch_size=1):
CaptumDerivative.__init__(self, trainer, n_steps=n_steps, internal_batch_size=internal_batch_size)
IntegratedGradients.__init__(self, self.trainer)
