leojklarner / gauche

A Library for Gaussian Processes in Chemistry
https://leojklarner.github.io/gauche/
MIT License
214 stars 22 forks source link

Add the "minmax" kernel #53

Open AustinT opened 1 year ago

AustinT commented 1 year ago

What does this implement/fix? Explain your changes

This PR provides a memory-efficient implementation for the minmax kernel from my personal BO repo.

It uses a formulation from Ioffe (2010) that allows torch.cdist to be used instead of directly computing mins/maxes.

What testing did you do to verify the changes in this PR?

None: unfortunately I could not get the environment to build properly on my machine so I could not run the tests. Maybe you can run them if you have a pre-built environment which works? I think the environment file in the main repository is out of date...

Pull Request Checklist

Ryan-Rhys commented 1 year ago

Thanks very much for adding this Austin! Will run the tests on my setup!

Ryan-Rhys commented 1 year ago

Apologies, I just revamped the fingerprint kernel classes in line with recent changes in GPyTorch. Merged over your MinMax implementation in a separate PR. Will add a note to the file that you authored the kernel in a bit!

AustinT commented 1 year ago

@Ryan-Rhys can you link me to the PR where this was merged? Happy to close this PR if the functionality has been added.