Skip to content

Commit

Permalink
Merge pull request #419 from JaxGaussianProcesses/cleanup_src
Browse files Browse the repository at this point in the history
Cleanup src
  • Loading branch information
thomaspinder authored Nov 29, 2023
2 parents e11a739 + c3aeedc commit 26aa40e
Show file tree
Hide file tree
Showing 35 changed files with 164 additions and 1,327 deletions.
11 changes: 7 additions & 4 deletions docs/examples/graph_kernels.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
# -*- coding: utf-8 -*-
# %% [markdown]
# # Graph Kernels
#
Expand All @@ -23,6 +22,7 @@
import matplotlib as mpl
import matplotlib.pyplot as plt
import networkx as nx
import optax as ox

with install_import_hook("gpjax", "beartype.beartype"):
import gpjax as gpx
Expand Down Expand Up @@ -154,11 +154,14 @@
# With a posterior defined, we can now optimise the model's hyperparameters.

# %%
opt_posterior, training_history = gpx.fit_scipy(
opt_posterior, training_history = gpx.fit(
model=posterior,
objective=jit(gpx.ConjugateMLL(negative=True)),
objective=gpx.ConjugateMLL(negative=True),
train_data=D,
)
optim=ox.adam(learning_rate=0.01),
num_iters=1000,
key=key
)

# %% [markdown]
#
Expand Down
308 changes: 0 additions & 308 deletions docs/examples/regression_mo.py

This file was deleted.

Loading

0 comments on commit 26aa40e

Please sign in to comment.