Python compile_to_logpdfの例

コード例 #1

ファイル: model.py プロジェクト: jeremiecoullon/mcx

 def logpdf(self, *args, **kwargs) -> float:
     """Compute the value of the distribution's logpdf."""
     logpdf, _, _, _ = compiler.compile_to_logpdf(self.graph,
                                                  self.namespace)
     return logpdf(*args, **kwargs)

コード例 #2

ファイル: model.py プロジェクト: jeremiecoullon/mcx

 def logpdf_src(self) -> str:
     """Return the source code of the log-probability density funtion
     generated by the compiler.
     """
     artifact = compiler.compile_to_logpdf(self.graph, self.namespace)
     return artifact.fn_source

コード例 #3

ファイル: sample.py プロジェクト: jeremiecoullon/mcx

def build_loglikelihood(model, **kwargs):
    artifact = compile_to_logpdf(model.graph, model.namespace)
    logpdf = artifact.compiled_fn
    loglikelihood = jax.partial(logpdf, **kwargs)
    return loglikelihood