def get_checkpoint_contribution()

in captum/influence/_core/tracincp_fast_rand_proj.py [0:0]

• 18 lines of code
• 1 McCabe index (conditional complexity)

        def get_checkpoint_contribution(checkpoint):

            assert (
                checkpoint is not None
            ), "None returned from `checkpoints`, cannot load."

            learning_rate = self.checkpoints_load_func(self.model, checkpoint)

            input_jacobians, input_layer_inputs = _basic_computation_tracincp_fast(
                self,
                inputs,
                targets,
            )

            src_jacobian, src_layer_input = _basic_computation_tracincp_fast(
                self, batch[0:-1], batch[-1]
            )
            return (
                _tensor_batch_dot(input_jacobians, src_jacobian)
                * _tensor_batch_dot(input_layer_inputs, src_layer_input)
                * learning_rate
            )