Add surrogate models on Discrete Lognormal (#1627)

* Add surrogate models on Discrete Lognormal * black
facebookresearch · Aug 28, 2024 · 6334cbc · 6334cbc
1 parent 16c72a8
commit 6334cbc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/nevergrad/optimization/optimizerlib.py b/nevergrad/optimization/optimizerlib.py
@@ -2135,6 +2135,12 @@ def __init__(
 RFMetaModelOnePlusOne = ParametrizedMetaModel(multivariate_optimizer=OnePlusOne, algorithm="rf").set_name(
     "RFMetaModelOnePlusOne", register=True
 )
+RFMetaModelLogNormal = ParametrizedMetaModel(
+    multivariate_optimizer=LognormalDiscreteOnePlusOne, algorithm="rf"
+).set_name("RFMetaModelLogNormal", register=True)
+NeuralMetaModelLogNormal = ParametrizedMetaModel(
+    multivariate_optimizer=LognormalDiscreteOnePlusOne, algorithm="neural"
+).set_name("NeuralMetaModelLogNormal", register=True)
 MetaModelPSO = ParametrizedMetaModel(multivariate_optimizer=PSO).set_name("MetaModelPSO", register=True)
 RFMetaModelPSO = ParametrizedMetaModel(multivariate_optimizer=PSO, algorithm="rf").set_name(
     "RFMetaModelPSO", register=True