Skip to content

Commit

Permalink
adding instructions for how to run deepensemble.py module
Browse files Browse the repository at this point in the history
  • Loading branch information
beckynevin committed Apr 1, 2024
1 parent 644f585 commit 8f89deb
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,13 @@ Getting a little more specific:

![python module overview](images/workflow_deepUQ.png)

These modules can be accessed via the ipython example notebooks or via the model modules (ie `DeepEnsemble.py`). For example, to ingest data and train a Deep Ensemble:
> cd src/scripts/
> python DeepEnsemble.py low 10 /Users/rnevin/Documents/DeepUQ/ --save_final_checkpoints=True --savefig=True --n_epochs=10
This command will train a 10 network, 10 epoch ensemble on the low noise data and will save figures and final checkpoints to the specified directory. For more information on the arguments:
> python DeepEnsemble.py --help
## Installation

### Clone this repo
Expand Down

0 comments on commit 8f89deb

Please sign in to comment.