jettjaniak / teren

Linking activation space features to model behavior
Apache License 2.0
0 stars 1 forks source link

make it harder to confuse feature_idx and feature_id #11

Closed jettjaniak closed 4 months ago

jettjaniak commented 4 months ago

I'm thinking we can use dicts with keys being feature ids everywhere, with the exception of tensors in SAEExamplesByFeature. But we can add a method to SAEExamplesByFeature that takes feature id and returns it's index. Or even have methods that give you a slice that you want based on feature id?