Skip to content

Commit

Permalink
update some docs for the emceesolver
Browse files Browse the repository at this point in the history
  • Loading branch information
raoulcollenteur committed Aug 11, 2023
1 parent 2d3af6d commit 9945c38
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 8 deletions.
4 changes: 2 additions & 2 deletions pastas/objective_functions.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""This module contains the objective functions.
"""This module contains the objective functions that can be used with the pastas
`EmceeSolve` solver.
"""
from numpy import pi, log
Expand All @@ -20,7 +21,6 @@ class GaussianLikelihood:
the residuals, and :math:`\\epsilon_i` is the residual at time :math:`i`. The
parameter :math:`\\sigma^2` need to be estimated.
"""

_name = "GaussianLikelihood"
Expand Down
10 changes: 4 additions & 6 deletions pastas/solver.py
Original file line number Diff line number Diff line change
Expand Up @@ -676,14 +676,14 @@ class EmceeSolve(BaseSolver):
nwalkers: int, optional
Number of walkers to use. Default is 20.
backend: emcee.backend, optional
One of the Backends from Emcee. See Emcee documentation for more
information.
One of the Backends from Emcee used to store MCMC results. See Emcee
for more information.
moves: emcee.moves, optional
The moves argument determines how the next step for a walker is chosen in
the MCMC approach. One of the Moves classes from Emcee has to be provided.
See Emcee documentation for more information.
parallel: bool, optional
Run the sampler in parallel or not. This requires the `emcee` package to
Run the sampler in parallel or not.
progress_bar: bool, optional
Show the progress bar or not. Requires the `tqdm` package to be installed.
**kwargs, optional
Expand All @@ -697,8 +697,6 @@ class EmceeSolve(BaseSolver):
>>> solver = ps.EmceeSolve(
... nwalkers=20,
... backend=ps.backends.HDFBackend("test.h5"),
... moves=ps.moves.StretchMove(),
... progress_bar=True,
... )
>>> ml.solve(solver=solver)
Expand Down Expand Up @@ -728,7 +726,7 @@ class EmceeSolve(BaseSolver):
emcee.EnsembleSampler
emcee.moves
emcee.backend
pastas.objectivefunctions
pastas.objective_functions
"""

Expand Down

0 comments on commit 9945c38

Please sign in to comment.