fit#

pymc_extras.inference.fit(method: str, **kwargs) → DataTree[source]#

Fit a model with an inference algorithm. See fit_pathfinder() and fit_laplace() for more details.

Parameters:

  • method (str) – Which inference method to run. Supported: pathfinder or laplace

  • kwargs (keyword arguments are passed on to the inference method.)

Return type:

DataTree