setu.to_numpyro#
- setu.to_numpyro(estimator, theta, x_observed, *, conditions=None, name='nle_likelihood')[source]#
Add NLE/MixedNLE log-likelihood as a factor in the active NumPyro model.
Accepts both NLE and MixedNLE – they have identical log_prob interfaces.
- Parameters:
estimator (
LikelihoodEstimator) – Trained NLE or MixedNLE instance.theta (
Array) – JAX array from numpyro.sample(), shape (theta_dim,).x_observed (
Array) – Observed data, shape (n_obs, x_dim) or (x_dim,).conditions (
Array|None) – Optional conditions, shape (n_obs, condition_dim).name (
str) – Name for the factor site.
- Return type: