Skip to content
New issue

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

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

Already on GitHub? # to your account

Allow to pass trainable inducing inputs to AbstractVariationalGaussian #485

Conversation

stefanocortinovis
Copy link
Contributor

Checklist

  • [ x ] I've formatted the new code by running hatch run dev:format before committing.
  • [ NA ] I've added tests for new code.
  • [ NA ] I've added docstrings for the new code.

Description

As things stand, AbstractVariationalGaussian defaults to static inducing_inputs and making them trainable requires manual surgery. This PR enables passing trainable (i.e. Real) or fixed (i.e. Static) inducing_inputs to AbstractVariationalGaussian. When an array is passed as inducing_inputs, it is wrapped in Real by default, to mimic the default behaviour in other GP libraries.

I am happy to write unit tests if required.

Issue Number: N/A

@thomaspinder thomaspinder merged commit 7c645c2 into JaxGaussianProcesses:main Nov 1, 2024
11 checks passed
@stefanocortinovis stefanocortinovis deleted the trainable_inducing_inputs branch November 1, 2024 15:52
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants