Skip to content

FlyingPumba/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

Contributing

Stars

1 star

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 99.3%
  • Dockerfile 0.7%