File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
optimas/generators/ax/developer Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -85,14 +85,13 @@ def get_MTGP(
85
85
device : torch .device = torch .device ("cpu" ),
86
86
dtype : torch .dtype = torch .double ,
87
87
) -> TorchModelBridge :
88
- """Instantiates a Multi-task Gaussian Process (MTGP) model that generates
89
- points with EI.
88
+ """Instantiate a Multi-task Gaussian Process (MTGP) model.
90
89
90
+ Points are generated with EI (Expected Improvement).
91
91
If the input experiment is a MultiTypeExperiment then a
92
92
Multi-type Multi-task GP model will be instantiated.
93
93
Otherwise, the model will be a Single-type Multi-task GP.
94
94
"""
95
-
96
95
if isinstance (experiment , MultiTypeExperiment ):
97
96
trial_index_to_type = {
98
97
t .index : t .trial_type for t in experiment .trials .values ()
You can’t perform that action at this time.
0 commit comments