Activity
add the option to have regularized posterior mean
add the option to have regularized posterior mean
add option that we don't learn kernel parameters and that the data st…
add option that we don't learn kernel parameters and that the data st…
fixed a bug that passed num_training_steps as a float
fixed a bug that passed num_training_steps as a float
integrated scheduling of num training steps as a function of data size
integrated scheduling of num training steps as a function of data size
converted to indexing in numpy instead of jax which is computationall…
converted to indexing in numpy instead of jax which is computationall…
changed clipping for std to be consistent
changed clipping for std to be consistent
fixed typing issue with jax PRNGKeyArray
fixed typing issue with jax PRNGKeyArray
Removed dependency on brax databuffer and implemented our own samplin…
Removed dependency on brax databuffer and implemented our own samplin…
removed scaning during training, enforced reinitialization of data bu…
removed scaning during training, enforced reinitialization of data bu…
adapted calibration to only calibrate epistemic uncertainty
adapted calibration to only calibrate epistemic uncertainty
add option to calibrate the bnn
add option to calibrate the bnn
added flag to calibrate only using epistemic uncertainty
added flag to calibrate only using epistemic uncertainty
added activation function as argument for models
added activation function as argument for models
restuctured code to minimize compilation during training
restuctured code to minimize compilation during training
fixes in datastats initialization for init function of bnn models
fixes in datastats initialization for init function of bnn models
Merge branch 'main' of github.com:lenarttreven/bayesian_statistical_m…
Merge branch 'main' of github.com:lenarttreven/bayesian_statistical_m…
updated interface to pass parameters to update function. Allows retra…
updated interface to pass parameters to update function. Allows retra…
defined input and output axis for each brnn model
defined input and output axis for each brnn model