search

TransformerLensOrg / CircuitsVis

Mechanistic Interpretability Visualizations using React
https://alan-cooney.github.io/CircuitsVis/
MIT License
192 stars 29 forks source link

Multi-sample text neuron activations #34

Closed danbraunai closed 1 year ago

danbraunai commented 1 year ago

Allows for showing multiple samples in the TextNeuronActivations vis, with pagination via select:

image

(Note, this PR uses functionality introduced in the topk PR. All topk commits are included here, but it would be cleaner to merge the topk PR first.)

danbraunai commented 1 year ago

Edit: Done in 6468f631d5

TODO: Put the colours from each sample in the whole dataset on the same scale. Currently each sample gets its own colour scale, which can be confusing when viewing activations across samples.

alan-cooney commented 1 year ago

Thanks for the PR!