lczerolens.lenses.gradient#

Compute Gradient heatmap for a given model and input.

Classes#

GradientLens

Class for gradient-based XAI methods.

Module Contents#

class lczerolens.lenses.gradient.GradientLens(*, input_requires_grad=True, **kwargs)[source]#

Bases: lczerolens.lens.Lens

Class for gradient-based XAI methods.

Parameters:

input_requires_grad (bool)

input_requires_grad = True[source]#
_intervene(model, **kwargs)[source]#

Intervene on the model.

Parameters:

model (LczeroModel) – The NNsight model.

Returns:

The intervention results.

Return type:

dict