Fix broken docs (#424)

* Fix broken docs * Fix kernel ref
JaxGaussianProcesses · Nov 30, 2023 · 55ecaac · 55ecaac
1 parent 07d99db
commit 55ecaac
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/docs/examples/barycentres.py b/docs/examples/barycentres.py
@@ -144,7 +144,7 @@ def fit_gp(x: jax.Array, y: jax.Array) -> tfd.MultivariateNormalFullCovariance:
 
     opt_posterior, _ = gpx.fit_scipy(
         model=posterior,
-        objective=gpx.ConjugateMLL(negative=True),
+        objective=gpx.objectives.ConjugateMLL(negative=True),
         train_data=D,
     )
     latent_dist = opt_posterior.predict(xtest, train_data=D)

diff --git a/docs/examples/graph_kernels.py b/docs/examples/graph_kernels.py
@@ -156,7 +156,7 @@
 # %%
 opt_posterior, training_history = gpx.fit(
     model=posterior,
-    objective=gpx.ConjugateMLL(negative=True),
+    objective=gpx.objectives.ConjugateMLL(negative=True),
     train_data=D,
     optim=ox.adam(learning_rate=0.01),
     num_iters=1000,

diff --git a/docs/examples/yacht.py b/docs/examples/yacht.py
@@ -167,7 +167,7 @@
 
 # %%
 n_train, n_covariates = scaled_Xtr.shape
-kernel = gpx.RBF(
+kernel = gpx.kernels.RBF(
     active_dims=list(range(n_covariates)),
     variance=np.var(scaled_ytr),
     lengthscale=0.1 * np.ones((n_covariates,)),
@@ -188,7 +188,7 @@
 # %%
 training_data = gpx.Dataset(X=scaled_Xtr, y=scaled_ytr)
 
-negative_mll = jit(gpx.ConjugateMLL(negative=True))
+negative_mll = jit(gpx.objectives.ConjugateMLL(negative=True))
 
 opt_posterior, history = gpx.fit_scipy(
     model=posterior,