diff --git a/docs/src/showcase/bayesian_neural_ode.md b/docs/src/showcase/bayesian_neural_ode.md index 0a981abd98e..abd4be4cf25 100644 --- a/docs/src/showcase/bayesian_neural_ode.md +++ b/docs/src/showcase/bayesian_neural_ode.md @@ -99,9 +99,9 @@ We sample using 500 warmup samples and 500 posterior samples. ```@example bnode integrator = Leapfrog(find_good_stepsize(h, p)) -prop = AdvancedHMC.NUTS{MultinomialTS, GeneralisedNoUTurn}(integrator) +kernel = HMCKernel(Trajectory{MultinomialTS}(integrator, GeneralisedNoUTurn())) adaptor = StanHMCAdaptor(MassMatrixAdaptor(metric), StepSizeAdaptor(0.45, integrator)) -samples, stats = sample(h, prop, p, 500, adaptor, 500; progress = true) +samples, stats = sample(h, kernel, p, 500, adaptor, 500; progress = true) ``` ## Step 5: Plot diagnostics