Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement a new model svgp in JAX #809

Open
5 tasks
PatrickOHara opened this issue Sep 14, 2022 · 0 comments
Open
5 tasks

Implement a new model svgp in JAX #809

PatrickOHara opened this issue Sep 14, 2022 · 0 comments
Assignees
Labels
New feature Adds a new feature

Comments

@PatrickOHara
Copy link
Collaborator

PatrickOHara commented Sep 14, 2022

@defaultobject has a repo called cleanair_aq_example that trains a JAX model on the LAQN data (see m_stgp.py).
We want to include the model into the cleanair repo.

  • Make sure importing the model from the stgp repo works
  • Create a new class that extends the ModelMixin class
  • Implement the 'fit' method
  • Implement the 'predict' method
  • make sure new class works with the other classes

.

@PatrickOHara PatrickOHara added the New feature Adds a new feature label Sep 14, 2022
@edasu edasu changed the title Implement a new model in JAX Implement a new model svgs in JAX Sep 21, 2022
@edasu edasu changed the title Implement a new model svgs in JAX Implement a new model svgp in JAX Sep 21, 2022
@edasu edasu mentioned this issue Oct 27, 2022
11 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
New feature Adds a new feature
Projects
None yet
Development

No branches or pull requests

3 participants