lczerolens.lenses.probing.probe#

Probing lens for XAI.

Attributes#

EPS

Classes#

Probe

Abstract class for probes.

SignalCav

Signal CAV probe.

Module Contents#

lczerolens.lenses.probing.probe.EPS = 1e-06[source]#
class lczerolens.lenses.probing.probe.Probe[source]#

Bases: abc.ABC

Abstract class for probes.

_trained = False[source]#
abstract train(activations, labels, **kwargs)[source]#

Train the probe.

Parameters:
  • activations (torch.Tensor)

  • labels (Any)

abstract predict(activations, **kwargs)[source]#

Predict with the probe.

Parameters:

activations (torch.Tensor)

class lczerolens.lenses.probing.probe.SignalCav[source]#

Bases: Probe

Signal CAV probe.

train(activations, labels, **kwargs)[source]#

Train the probe.

Parameters:
  • activations (torch.Tensor)

  • labels (torch.Tensor)

predict(activations, **kwargs)[source]#

Predict with the probe.

Parameters:

activations (torch.Tensor)