search

IBM / UQ360

Uncertainty Quantification 360 (UQ360) is an extensible open-source toolkit that can help you estimate, communicate and use uncertainty in machine learning model predictions.
https://uq360.res.ibm.com/
Apache License 2.0
257 stars 61 forks source link

Support dataloader in BNNRegression #22

Open pronics2004 opened 2 years ago

pronics2004 commented 2 years ago

Update BNNRegression to support iterating over dataloader during fit similar to BNNClassification.

https://github.com/IBM/UQ360/blob/main/uq360/algorithms/variational_bayesian_neural_networks/bnn.py#L54

https://github.com/IBM/UQ360/blob/main/uq360/algorithms/variational_bayesian_neural_networks/bnn.py#L216