diff --git a/surfaces/__init__.py b/surfaces/__init__.py index 0a8c938..22361ef 100644 --- a/surfaces/__init__.py +++ b/surfaces/__init__.py @@ -5,11 +5,20 @@ __version__ = "0.1.0" __license__ = "MIT" -from .mathematical_functions import mathematical_functions +from .mathematical_functions import ( + mathematical_functions, + mathematical_functions_1d, + mathematical_functions_2d, + mathematical_functions_nd, +) + from .machine_learning_functions import machine_learning_functions __all__ = [ "mathematical_functions", + "mathematical_functions_1d", + "mathematical_functions_2d", + "mathematical_functions_nd", "machine_learning_functions", ] diff --git a/surfaces/mathematical_functions/__init__.py b/surfaces/mathematical_functions/__init__.py index 7e2fb88..e1ff2f8 100644 --- a/surfaces/mathematical_functions/__init__.py +++ b/surfaces/mathematical_functions/__init__.py @@ -81,3 +81,39 @@ StyblinskiTangFunction, ThreeHumpCamelFunction, ] + + +mathematical_functions_1d = [ + GramacyAndLeeFunction, +] + + +mathematical_functions_2d = [ + AckleyFunction, + BealeFunction, + BoothFunction, + BukinFunctionN6, + CrossInTrayFunction, + DropWaveFunction, + EasomFunction, + EggholderFunction, + GoldsteinPriceFunction, + HimmelblausFunction, + HölderTableFunction, + LangermannFunction, + LeviFunctionN13, + MatyasFunction, + McCormickFunction, + RosenbrockFunction, + SchafferFunctionN2, + SimionescuFunction, + ThreeHumpCamelFunction, +] + + +mathematical_functions_nd = [ + GriewankFunction, + RastriginFunction, + SphereFunction, + StyblinskiTangFunction, +]