def __init__(self, predictor: Predictor):

        CaptumAttribution.__init__(self, 'deeplift', predictor)

        self.submodel = self.predictor._model.captum_sub_model()
        DeepLift.__init__(self, self.submodel)
Beispiel #2
0
 def __init__(self, trainer):
     CaptumDerivative.__init__(self, trainer)
     DeepLift.__init__(self, self.trainer)