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)