Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Tuned Lens
Tuned Lens

API Reference

  • tuned_lens.nn.lenses
  • tuned_lens.nn.unembed
  • tuned_lens.plotting
    • tuned_lens.plotting.prediction_trajectory
    • tuned_lens.plotting.token_formatter
    • tuned_lens.plotting.trajectory_plotting
  • tuned_lens.load_artifacts

Tutorials

  • Loading a pre-trained lens
  • Training and evaluating lenses
  • Comparing prediction trajectories
  • Combing the Tuned Lens and the Transformer Lens
  • Maintainers Guide
Back to top
View this page
Edit this page

tuned_lens.plotting¶

Provides tools for plotting.

Modules

prediction_trajectory

Plot a lens table for some given text and model.

token_formatter

Contains a class for formatting tokens for display in plots.

trajectory_plotting

Contains utility classes for creating heatmap visualizations.

Next
tuned_lens.plotting.prediction_trajectory
Previous
tuned_lens.nn.unembed
Copyright © 2023, FAR AI
Made with Sphinx and @pradyunsg's Furo