Skip to content
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.

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

Add MDN with support for parameter transformations #7

Open
5 tasks
alvorithm opened this issue Mar 5, 2020 · 0 comments
Open
5 tasks

Add MDN with support for parameter transformations #7

alvorithm opened this issue Mar 5, 2020 · 0 comments

Comments

@alvorithm
Copy link

Building on Marcel's work, add an MDN (Mixture Density Network) implementation that can play well with e.g. SNPE-A proposal posterior corrections.

This is the wishlist

  • basic MDN that works and is useable from sbi (mixture of full-covariance Gaussians)
  • MDN becomes highly compositional - only last layer implemented (MDN layer), the rest is built e.g. using Sequential and possibly heuristics given data dimensions, etc.
  • MDN layer returns a MixtureSameFamily density (becoming part of PyTorch, PR)
  • (long term) MDN layer can self-configure given the particulars of the desired MixtureSameFamily
  • Building the whole MDN does not require specification of redundant information, much like Keras s sequential (or look at thinc for a more functional take).
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

No branches or pull requests

1 participant