Skip to main content
Ctrl+K
lczerolens - Home lczerolens - Home
  • Getting Started
  • Features
  • Tutorials
  • API Reference
  • About
  • GitHub
  • Discord
  • PyPI
  • Getting Started
  • Features
  • Tutorials
  • API Reference
  • About
  • GitHub
  • Discord
  • PyPI

Section Navigation

  • lczerolens
    • lczerolens.backends
    • lczerolens.board
    • lczerolens.concept
    • lczerolens.concepts
    • lczerolens.constants
    • lczerolens.lens
    • lczerolens.lenses
    • lczerolens.model
    • lczerolens.play
  • API Reference
  • lczerolens
  • lczerolens.lenses
  • lczerolens.lenses.sae
  • lczerolens.lenses.sae.buffer

lczerolens.lenses.sae.buffer#

Activation lens for XAI.

Classes#

ActivationBuffer

Module Contents#

class lczerolens.lenses.sae.buffer.ActivationBuffer[source]#
model: lczerolens.model.LczeroModel[source]#
dataset: datasets.Dataset[source]#
compute_fn: Callable[[Any, lczerolens.model.LczeroModel], torch.Tensor][source]#
n_batches_in_buffer: int = 10[source]#
compute_batch_size: int = 64[source]#
train_batch_size: int = 2048[source]#
dataloader_kwargs: dict | None = None[source]#
logger: Callable | None = None[source]#
__post_init__()[source]#
_make_dataloader_it()[source]#
_fill_buffer()[source]#
_make_activations_it()[source]#
__iter__()[source]#
__next__()[source]#

previous

lczerolens.lenses.sae

next

lczerolens.model

On this page
  • Classes
  • Module Contents
    • ActivationBuffer
      • ActivationBuffer.model
      • ActivationBuffer.dataset
      • ActivationBuffer.compute_fn
      • ActivationBuffer.n_batches_in_buffer
      • ActivationBuffer.compute_batch_size
      • ActivationBuffer.train_batch_size
      • ActivationBuffer.dataloader_kwargs
      • ActivationBuffer.logger
      • ActivationBuffer.__post_init__()
      • ActivationBuffer._make_dataloader_it()
      • ActivationBuffer._fill_buffer()
      • ActivationBuffer._make_activations_it()
      • ActivationBuffer.__iter__()
      • ActivationBuffer.__next__()

© Copyright 2024, Yoann Poupart.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.16.1.