Skip to content

feat(uq-models): implement deep ensemble and MC dropout models, add UQ visualizations to PostProcessor. #85

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
khakis23 wants to merge 33 commits into
base: develop
Choose a base branch
from
Open

feat(uq-models): implement deep ensemble and MC dropout models, add UQ visualizations to PostProcessor. #85

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
c9cab3e
feat: implement deep enesemble model and test, add NLL loss function
Jun 30, 2026
6ad02b5
feat: create logic for DE visualizations, integrate diag_val_plot
Jul 3, 2026
9045826
refactor: break out uncertainty check and scale logic
Jul 3, 2026
a79ec76
feat: integrate validation_plot with DE's
Jul 3, 2026
0e79874
feat: override fit for DE, integrate nn_learning_plot with DE
Jul 3, 2026
757c6d8
bug: remove override decorator and import
Jul 3, 2026
95b2313
bug: add checks in DE hypermodel fit method override
Jul 3, 2026
4b25e6f
feat: add DE-specific visualiztion:
Jul 7, 2026
e277106
bug(PostP.ensemble_uncertainty_plot): warn user about model retrain o…
Jul 7, 2026
c93043a
docs: (Grant) add docs to DE tests
Jul 7, 2026
5758909
bug: fix multi-dimentional error when plotting validation plot with u…
Jul 12, 2026
2d4ff57
tests: change model shape assetion from (6,6) to (6,8)
Jul 12, 2026
0f79330
feat: add DE's to 3 notebooks
Jul 12, 2026
bf41a22
docs: add DE to notebook markdown cells
Jul 15, 2026
badf8cc
docs: add parameters to method docstring
Jul 15, 2026
70c002e
chore: remove debugging print statement
Jul 18, 2026
9ebb9e9
chore: remove debugging print statement...again
Jul 18, 2026
59c6aa5
WIP: add MCDropoutLayer for Monte Carlo Dropout UQ
Grspiegel Jul 22, 2026
62a7d81
Merge remote-tracking branch 'upstream/deep-ensemble' into deep-ensemble
Grspiegel Jul 22, 2026
638ec8f
reafactor/feat: move DE visualization out of PostProcessor into new c…
Jul 22, 2026
127a706
bug: fix syntax error
Jul 22, 2026
65329f2
refactor: abstract UQ further out of PostP
Jul 22, 2026
76a3294
feat: complete UQ-style wrapper arch for MC dropout
Jul 26, 2026
a2c2013
refactor: condense UQ instantiation on Tuner init
Jul 26, 2026
168634e
Merge: upstream deep-ensemble into fork
Jul 26, 2026
bce42ff
chore: update gitignore
Jul 26, 2026
9ab60b8
bug: fix skorch disabling dropout during UQ predictions for MCD
Jul 28, 2026
767b148
bug: add check for prediction type
Jul 29, 2026
7b5213f
bug: fix validation plot not showing variance for all points
Aug 1, 2026
72a8716
refactor: update how UQVis works with PostProcessor
Aug 1, 2026
76b69c1
feat WIP: add notebook
Aug 6, 2026
bb643bf
feat: complete UQ notebook
Aug 15, 2026
694bd6f
docs: change UQ notebook nn_learning_plot markdown to reference BWR i…
Aug 15, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -135,3 +135,6 @@ dmypy.json
# Pyre type checker
.pyre/
.claude/

# IDE
.idea
10 changes: 10 additions & 0 deletions .idea/.gitignore
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

89 changes: 89 additions & 0 deletions .idea/inspectionProfiles/Project_Default.xml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 6 additions & 0 deletions .idea/inspectionProfiles/profiles_settings.xml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

18 changes: 18 additions & 0 deletions .idea/misc.xml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

8 changes: 8 additions & 0 deletions .idea/modules.xml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

17 changes: 17 additions & 0 deletions .idea/pyMAISEgrant.iml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 6 additions & 0 deletions .idea/vcs.xml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading