Skip to content

cybershiptrooper/iit

Repository files navigation

Interchange Intervention Training

A replication of the paper "Inducing Causal Structure for Interpretable Neural Networks" by Geiger et al.

Installation

You can use the Dockerfile to set it up. Alternatively, you can use poetry install from the root folder.

Running

There is only one task you can run this on for now.

Running python train.py trains a model using Intercahnge Interventions + Multi Task (1)

Running python eval.py generates plots for how accurate the circuit induced is.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •