evanhanders / iit

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


Languages

Language:Python 99.2%Language:Dockerfile 0.8%