gwinferno.pipeline.analysis.find_map#
- find_map(rng_key, numpyro_model, *model_args, Niter=100, lr=0.01)[source]#
Find the MAP estimate for a given NumPyro model using SVI with Adam optimizing the ELBO
- Parameters:
- rng_keyjax.random.PRNGKey
RNG Key to be passed to SVI.run().
- numpyro_modelcallable
Python callable containing numpyro.primitives.
- Niterint, optional
Number of iterations to run variational inference. Defaults to 100.
- lrfloat, optional
learning rate used for Adam optimizer. Defaults to 0.01.
- Returns:
- SVIRunResult.paramsdict
parameters of the result of MAP optimization