1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
import numpy as np
class Function(object):
def __init__(self, f=None, limits=None, start=None):
"""
f=callable is the function
limits=(lo,hi) are the limits
start=vector is the start point
"""
self.f, self.limits, self.start = f, limits, start
self.__name__ = f.__name__
def __call__(self, p):
return self.f(p)
def response_surface(self, p=None, dims=[0, 1]):
if p is None:
p = self.start
plot_response_surface(self, p, dims)
class Fitness(object):
def __init__(self, f=None, data=None, limits=None, start=None):
self.f, self.limits, self.start = f, limits, start
if len(data) == 2:
self.x, self.y = data
self.dy = None
else:
self.x, self.y, self.dy = data
def profile(self, p):
return self.x, self.f(self.x, p)
def residuals(self, p):
return (self.profile(p) - self.y) / self.dy
def __call__(self, p):
return np.sum(self.residuals(p) ** 2)
def plot(self, p=None):
"""
Plot a profile for the given p
"""
import pylab
if self.dy is not None:
pylab.errorbar(self.x, self.y, yerr=self.dy, fmt="x")
else:
pylab.plot(self.x, self.y, "x")
if p is None:
p = self.start
x, y = self.profile(p)
pylab.plot(x, y)
def response_surface(self, p=None, dims=[0, 1]):
if p is None:
p = self.start
plot_response_surface(self, p, dims)
def plot_response_surface(f, p, dims=[0, 1]):
"""
Plot a line or a slice around a point in a n-D function
"""
import pylab
if len(dims) == 1:
xi = dims[0]
x = pylab.linspace(-10, 10, 40) - p[xi]
def value(v):
p[xi] = v
return f(p)
z = [value(v) for v in x]
pylab.plot(x, z)
else:
xi, yi = dims
x = pylab.linspace(-10, 10, 40) - p[xi]
y = pylab.linspace(-10, 10, 40) - p[yi]
def value(pt):
p[xi] = pt[0]
p[yi] = pt[1]
return f(p)
z = np.array([[value((v, w)) for v in x] for w in y])
pylab.pcolor(x, y, z)
|
