Skip to content

Commit

Permalink
Update NUTS docs to reflect integrators being gone (#3589)
Browse files Browse the repository at this point in the history
* Update NUTS docs to reflect integrators being gone

* Typo fix
  • Loading branch information
ColCarroll authored and rpgoldman committed Aug 15, 2019
1 parent c1b9272 commit 58bed9f
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions pymc3/step_methods/hmc/nuts.py
Expand Up @@ -121,14 +121,10 @@ def __init__(self, vars=None, max_treedepth=10, early_max_treedepth=8,
Whether step size adaptation should be enabled. If this is
disabled, `k`, `t0`, `gamma` and `target_accept` are ignored.
max_treedepth : int, default=10
The maximum tree depth. Trajectories are stoped when this
The maximum tree depth. Trajectories are stopped when this
depth is reached.
early_max_treedepth : int, default=8
The maximum tree depth during the first 200 tuning samples.
integrator : str, default "leapfrog"
The integrator to use for the trajectories. One of "leapfrog",
"two-stage" or "three-stage". The second two can increase
sampling speed for some high dimensional problems.
scaling : array_like, ndim = {1,2}
The inverse mass, or precision matrix. One dimensional arrays are
interpreted as diagonal matrices. If `is_cov` is set to True,
Expand Down

0 comments on commit 58bed9f

Please sign in to comment.