Hypervolume Knowledge Gradient (#1950)

Summary: Pull Request resolved: #1950 This includes support for decoupled evaluations and multi-fidelity optimization Reviewed By: Balandat Differential Revision: D47809667 fbshipit-source-id: a2115f78c987be56bb1e68ddd3850d736b2bb08c
pytorch · Aug 14, 2023 · 9ea65de · 9ea65de
1 parent 8aa2122
commit 9ea65de
Show file tree

Hide file tree

Showing 5 changed files with 1,007 additions and 0 deletions.
diff --git a/botorch/acquisition/multi_objective/__init__.py b/botorch/acquisition/multi_objective/__init__.py
@@ -8,6 +8,10 @@
     ExpectedHypervolumeImprovement,
     MultiObjectiveAnalyticAcquisitionFunction,
 )
+from botorch.acquisition.multi_objective.hypervolume_knowledge_gradient import (
+    qHypervolumeKnowledgeGradient,
+    qMultiFidelityHypervolumeKnowledgeGradient,
+)
 from botorch.acquisition.multi_objective.max_value_entropy_search import (
     qMultiObjectiveMaxValueEntropy,
 )
@@ -35,6 +39,8 @@
     "get_default_partitioning_alpha",
     "prune_inferior_points_multi_objective",
     "qExpectedHypervolumeImprovement",
+    "qHypervolumeKnowledgeGradient",
+    "qMultiFidelityHypervolumeKnowledgeGradient",
     "qNoisyExpectedHypervolumeImprovement",
     "MOMF",
     "qMultiObjectiveMaxValueEntropy",