Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
melanibe authored Mar 21, 2024
1 parent 1c5b505 commit 46b355a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion causal_models/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ The code in this folder is adapted from the official code associated with the
## Train the counterfactual inference model

To train the counterfactual inference models in the paper you can simply run
`python causal_models/main.py --hps embed` replace by `padchest` if you want to train on chest x-rays. All associated hyperparameters are stored in `causal_models/hps.py`.
`python causal_models/main.py --hps embed --exp_name your_exp_name` replace by `padchest` if you want to train on chest x-rays. All associated hyperparameters are stored in `causal_models/hps.py`.

This assumes you have already set up your data folders as per the main repository.

Expand Down

0 comments on commit 46b355a

Please sign in to comment.