ramsey-devs / ramsey

Probabilistic deep learning using JAX
https://ramsey.rtfd.io
Apache License 2.0
13 stars 3 forks source link

DRAFT: First basic implementation of GP #9

Closed mamarder closed 1 year ago

mamarder commented 1 year ago

Overview A first draft of a simple GP is implemented in ramsey/_src/gaussian_process/gaussian_process.py

An RBF Kernel is offered in ramsey/_src/gaussian_process/kernel.py In general the design is chosen so that a library consumer can implement own kernels.

An example on how to use the the module and some visualization is given in examples/gaussian_process/gaussian_process.py This examples generates plots as shown below.

image image

The example generates data by using function from examples/gaussian_process/data.py

Limitations

dirmeier commented 1 year ago

Thanks! Looks great, I'll merge this now.