2de1df1704c1c187c72aacc15efcacd7029139dd,stellargraph/utils/saliency_maps/integrated_gradients.py,IntegratedGradients,__init__,#IntegratedGradients#Any#Any#,40

Before Change

    

    def __init__(self, model, generator):
        super().__init__(model, generator)

    def get_integrated_node_masks(
        self,
        node_idx,

After Change

            self._adj = generator.A_values
            self._adj_inds = generator.A_indices
        else:
            if not isinstance(generator, FullBatchSequence):
                raise TypeError(
                    "The generator supplied has to be an object of FullBatchSequence for dense adjacency matrix."
                )
            if len(model.input) != 3:
                raise RuntimeError(
                    "Keras model for dense adjacency is expected to have three inputs"
                )

            self._adj = generator.A_dense

        // Extract features from generator

Instances