def __init__(self, predictor: Predictor): CaptumAttribution.__init__(self, 'deeplift', predictor) self.submodel = self.predictor._model.captum_sub_model() DeepLift.__init__(self, self.submodel)
def __init__(self, trainer): CaptumDerivative.__init__(self, trainer) DeepLift.__init__(self, self.trainer)