(ax: Axes, kernel: gpflow.kernels.Kernel)
| 345 | |
| 346 | # %% |
| 347 | def plot_2d_kernel_samples(ax: Axes, kernel: gpflow.kernels.Kernel) -> None: |
| 348 | n_grid = 30 |
| 349 | X = np.zeros((0, 2)) |
| 350 | Y = np.zeros((0, 1)) |
| 351 | model = gpflow.models.GPR((X, Y), kernel=deepcopy(kernel)) |
| 352 | |
| 353 | Xplots = np.linspace(-0.6, 0.6, n_grid) |
| 354 | Xplot1, Xplot2 = np.meshgrid(Xplots, Xplots) |
| 355 | Xplot = np.stack([Xplot1, Xplot2], axis=-1) |
| 356 | Xplot = Xplot.reshape([n_grid ** 2, 2]) |
| 357 | |
| 358 | tf.random.set_seed(20220903) |
| 359 | fs = model.predict_f_samples(Xplot, num_samples=1) |
| 360 | fs = fs.numpy().reshape((n_grid, n_grid)) |
| 361 | ax.plot_surface(Xplot1, Xplot2, fs, cmap=coolwarm) |
| 362 | ax.set_title("Example $f$") |
| 363 | |
| 364 | |
| 365 | def plot_2d_kernel_prediction(ax: Axes, kernel: gpflow.kernels.Kernel) -> None: |
no test coverage detected
searching dependent graphs…