From 5a01e63264d90c6a04d1d97251dff131c8db7b67 Mon Sep 17 00:00:00 2001 From: Daniel Dodd Date: Wed, 7 Sep 2022 09:43:20 +0100 Subject: [PATCH] Update classification.ipynb --- examples/classification.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/classification.ipynb b/examples/classification.ipynb index 32cf9fa12..749cfb925 100644 --- a/examples/classification.ipynb +++ b/examples/classification.ipynb @@ -421,7 +421,7 @@ "\n", "mll = jax.jit(posterior.marginal_log_likelihood(D, constrainer, negative=False))\n", "\n", - "adapt = blackjax.window_adaptation(blackjax.nuts, mll, num_adapt, target_acceptance_rate=0.65)\n", + "adapt = blackjax.window_adaptation(blackjax.nuts, unconstrained_mll, num_adapt, target_acceptance_rate=0.65)\n", "\n", "# Initialise the chain\n", "last_state, kernel, _ = adapt.run(key, params)\n",