We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent da6da9a commit 9298222Copy full SHA for 9298222
1 file changed
nevergrad/optimization/metamodel.py
@@ -174,7 +174,8 @@ def my_kernel(x, y):
174
DiscreteLenglerOnePlusOne = registry["DiscreteLenglerOnePlusOne"]
175
176
def loss_function_sm(x):
177
- return float(model.predict(trans(np.asarray(x, dtype=X[0].dtype).flatten()[None, :])))
+ x = np.asarray(x, dtype=X[0].dtype).flatten()[None, :]
178
+ return model.predict(trans(x)).item()
179
180
for cls in (
181
(Powell, DE) if algorithm != "image" else (DiscreteLenglerOnePlusOne,)
0 commit comments