Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix qNEI with Derivative Enabled BO (#1716)
Summary: <!-- Thank you for sending the PR! We appreciate you spending the time to make BoTorch better. Help us understand your motivation by explaining why you decided to make this change. You can learn more about contributing to BoTorch here: https://github.com/pytorch/botorch/blob/main/CONTRIBUTING.md --> ## Motivation Derivative enabled GPs allow for faster convergence rates. This is well studied in the literature and makes intuitive sense since a lot more information is provided with each objective function evaluation. I want code to be available for myself and other future researchers who want to explore derivative enabled GPs. This PR resolves an error with the qNEI acquisition function for derivative enabled GPs. For more information, see [here](#1679). ### Have you read the [Contributing Guidelines on pull requests](https://github.com/pytorch/botorch/blob/main/CONTRIBUTING.md#pull-requests)? Yes, this doesn't affect the outcome of the unit tests, though when I run the unit tests on an unmodified branch of *main* I get `110 failed, 853 passed, 1165 warnings`. Pull Request resolved: #1716 Test Plan: I have some files from an issue ([1679](#1679)) I opened which show a simple case where a derivative-enabled Gaussian Process fails to find the qNEI value at a point where an equivalent non-derivative-enabled Gaussian Process does not fail. - [shared.py.txt](https://github.com/pytorch/botorch/files/10820140/shared.py.txt) - [Simple_Repro_non_D_2.py.txt](https://github.com/pytorch/botorch/files/10820141/Simple_Repro_non_D_2.py.txt) - [Simple_Repro_non_D.py.txt](https://github.com/pytorch/botorch/files/10820142/Simple_Repro_non_D.py.txt) Adding the fix in this PR removes the error. Please let me know if I should add a specific unit test for this. ## Related PRs N/A Reviewed By: SebastianAment Differential Revision: D43822902 Pulled By: esantorella fbshipit-source-id: 0876bebb9364fca6c10379d92e792cb2dc9865e1
- Loading branch information