Skip to content

MDN VAE - #6

Merged
nickwimer merged 6 commits into
mainfrom
vae_test
Jun 5, 2026
Merged

MDN VAE#6
nickwimer merged 6 commits into
mainfrom
vae_test

Conversation

@nickwimer

Copy link
Copy Markdown
Collaborator

No description provided.

* Bayesian NNs (#3)

* adding support for hyperparameter tuning using Ray

* remove manditory r2 and rmse from plot text

* WIP; initial code for ptBNN replicating tf flipout

* adding in updates to BNN working tests for iaps...

* adjusting the clamping for the MDN output to try to avoid NaNs

* updates to MDN output for stability

* adding in seed for reproduction testing

* cleaning up before PR

* fixing typos and adding in conditions for partial bayes blocks

* removing pass for unsupported architectures...todo to fully implement

* RNN models and LR Schedulers (#4)

* moving utility functions into class files...might deprecate soon

* making the mixture density loss have mse regularization

* adding in schedulers and early stopping

* fixing mse addition loss term for MDNs

* updating regression notebook

* adding in support for LSTM model from time series modeling work

* cleaning up old commented code

* removing commented code
@nickwimer
nickwimer merged commit 313b7b9 into main Jun 5, 2026
1 check passed
@nickwimer
nickwimer deleted the vae_test branch June 5, 2026 20:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant