ENH: Bayesian neural network architectures with training requisites #126

Aakanksha-Rana · 2021-03-22T17:28:55Z

Types of changes

Bug fix (non-breaking change which fixes an issue)
[ x] New feature (non-breaking change which adds functionality)
Breaking change (fix or feature that would cause existing functionality to change)

Summary

Checklist

All tests passing
I have added tests to cover my changes
I have updated documentation (if necessary)
My code follows the code style of this project
(we are using black: you can pip install pre-commit,
run pre-commit install in the pydra directory
and black will be run automatically with each commit)

Acknowledgment

I acknowledge that this contribution will be available under the Apache 2 license.

satra · 2021-03-25T00:43:13Z

please add some tests, to make sure the code does what it should do. this is not realistic training, but just to exercise the code on some dummy but predefined data.

Aakanksha-Rana · 2021-03-25T10:45:30Z

Thanks. I will add them.

Bayesian decoder

full bayesian Vnet

Aakanksha-Rana · 2021-05-20T21:01:11Z

Adding new architectures. Tests will be added soon for all of them.

for more information, see https://pre-commit.ci

adapted for multi_class_segmentation

for more information, see https://pre-commit.ci

Updated for multi-class

for more information, see https://pre-commit.ci

Aakanksha-Rana · 2021-06-19T17:28:30Z

Bayesian Training is updated now with tests.

nobrainer/models/asdnet.py

nobrainer/models/bayesian_vnet.py

nobrainer/models/bayesian-meshnet.py

nobrainer/models/bayesian_vnet_semi.py

nobrainer/models/vnet.py

satra · 2021-06-19T18:37:13Z

nobrainer/models/vnet.py

+
+def vnet(
+    n_classes=1,
+    input_shape=(256, 256, 256, 1),


@Aakanksha-Rana - does this actually work at this resolution?

Aakanksha-Rana · 2021-10-22T14:50:38Z

Fixes #104

Aakanksha-Rana · 2021-10-22T15:08:28Z

Fixes #107

Aakanksha-Rana and others added 3 commits November 18, 2020 14:51

asdnet_added

8573fdd

Create bayesian_utils.py

d33c2d0

Create bayesian-meshnet.py

f289a3e

Aakanksha-Rana changed the title ~~[WIP] Bayesian Neural Network Training (Meshnet + UNet)~~ [WIP] Bayesian Neural Network Architectures with training requisites Mar 23, 2021

Aakanksha-Rana changed the title ~~[WIP] Bayesian Neural Network Architectures with training requisites~~ [WIP] bayesian neural network architectures with training requisites Mar 23, 2021

Merge branch 'master' into bayesian-training

0738d29

Aakanksha-Rana added 3 commits May 20, 2021 16:43

Create bayesian_vnet_semi.py

2d79c20

Bayesian decoder

Create bayesian_vnet.py

5deedf8

full bayesian Vnet

Create vnet.py

8766a3d

satra and others added 18 commits June 13, 2021 11:30

Merge branch 'master' into bayesian-training

39531dd

[pre-commit.ci] auto fixes from pre-commit.com hooks

20a8930

for more information, see https://pre-commit.ci

Merge branch 'master' into bayesian-training

89a12ab

fix: reorder positional arg

27588c6

[pre-commit.ci] auto fixes from pre-commit.com hooks

c47f728

for more information, see https://pre-commit.ci

Merge branch 'master' into bayesian-training

39f22f2

Update vnet.py

7748f34

Update models_test.py

e1e9442

Update vnet.py

ff96c14

[pre-commit.ci] auto fixes from pre-commit.com hooks

215d2ff

for more information, see https://pre-commit.ci

Update bayesian_vnet_semi.py

62b4ab2

adapted for multi_class_segmentation

[pre-commit.ci] auto fixes from pre-commit.com hooks

5894d85

for more information, see https://pre-commit.ci

Update bayesian_vnet_semi.py

649f4db

Update bayesian_utils.py

d7f5eb4

[pre-commit.ci] auto fixes from pre-commit.com hooks

dc450bf

for more information, see https://pre-commit.ci

Update bayesian_vnet_semi.py

9299fb5

[pre-commit.ci] auto fixes from pre-commit.com hooks

5ddb2e1

for more information, see https://pre-commit.ci

Update bayesian_vnet.py

7073c84

Updated for multi-class

pre-commit-ci bot and others added 14 commits June 19, 2021 16:19

[pre-commit.ci] auto fixes from pre-commit.com hooks

e97c1ad

for more information, see https://pre-commit.ci

Update models_test.py

2a0100d

[pre-commit.ci] auto fixes from pre-commit.com hooks

c7c161a

for more information, see https://pre-commit.ci

Update models_test.py

8692b3e

Update models_test.py

28ed033

[pre-commit.ci] auto fixes from pre-commit.com hooks

18a5753

for more information, see https://pre-commit.ci

Update bayesian_vnet_semi.py

c671a2a

[pre-commit.ci] auto fixes from pre-commit.com hooks

ed220ab

for more information, see https://pre-commit.ci

Update bayesian_utils.py

2ba38f8

[pre-commit.ci] auto fixes from pre-commit.com hooks

8eeda35

for more information, see https://pre-commit.ci

Update bayesian-meshnet.py

b4777ca

Update bayesian_utils.py

f40ac54

[pre-commit.ci] auto fixes from pre-commit.com hooks

313685e

for more information, see https://pre-commit.ci

Update bayesian_utils.py

11ddbce

Aakanksha-Rana changed the title ~~[WIP] bayesian neural network architectures with training requisites~~ Bayesian neural network architectures with training requisites Jun 19, 2021

Merge branch 'master' into bayesian-training

8660197