39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from typing import Callable, Tuple
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Function:
|
||
|
|
xlim: Tuple[float, float]
|
||
|
|
ylim: Tuple[float, float]
|
||
|
|
minimum: Tuple[float, float]
|
||
|
|
eval: Callable
|
||
|
|
|
||
|
|
def grid(self, x_dim=256, y_dim=256):
|
||
|
|
x = np.linspace(self.xlim[0], self.xlim[1], x_dim)
|
||
|
|
y = np.linspace(self.ylim[0], self.ylim[1], y_dim)
|
||
|
|
X, Y = np.meshgrid(x, y)
|
||
|
|
return x, y, self.eval(X, Y)
|
||
|
|
|
||
|
|
class Rastrigin(Function):
|
||
|
|
def __init__(self, A: int = 10):
|
||
|
|
super().__init__(
|
||
|
|
xlim=(-5.12, 5.12),
|
||
|
|
ylim=(-5.12, 5.12),
|
||
|
|
minimum=(0, 0),
|
||
|
|
eval=lambda x, y: self.A * 2 + \
|
||
|
|
(x**2 - self.A * np.cos(2 * np.pi * x)) + \
|
||
|
|
(y**2 - self.A * np.cos(2 * np.pi * y)))
|
||
|
|
self.A = A
|
||
|
|
|
||
|
|
class Sphere(Function):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__(
|
||
|
|
xlim=(-5, 5),
|
||
|
|
ylim=(-5, 5),
|
||
|
|
minimum=(0, 0),
|
||
|
|
eval=lambda x, y: x**2 + y**2)
|
||
|
|
|