Update classification.ipynb

JaxGaussianProcesses · Sep 16, 2022 · 5a01e63 · 5a01e63
1 parent fe1cf90
commit 5a01e63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",