我认为你必须为此推出自己的近似值。这个想法很简单:在一些合理的点对函数进行采样(至少与泰勒近似中的单项式一样多,但最好更多),并用 拟合系数np.linalg.lstsq
。实际合身是一条线,剩下的就是为它做准备。
我将使用 n=3 和 m=2 的示例,即三个变量和二维值。初始设置:
import numpy as np
def f(point):
x, y, z = point[0], point[1], point[2]
return np.array([np.exp(x + 2*y + 3*z), np.exp(3*x + 2*y + z)])
n = 3
m = 2
scale = 0.1
参数的选择与(参见源代码scale
)的文档字符串中的考虑相同。approximate_taylor_polynomial
下一步是生成点。对于 n 个变量,二次拟合涉及1 + n + n*(n+1)/2
单项式(一个常数,n 线性,n(n+1)/2 二次)。我使用1 + n + n**2
放置在周围(0, 0, 0)
并具有一个或两个非零坐标的点。特定的选择有些武断。我找不到多元二次拟合的样本点的“规范”选择。
points = [np.zeros((n, ))]
points.extend(scale*np.eye(n))
for i in range(n):
for j in range(n):
point = np.zeros((n,))
point[i], point[j] = scale, -scale
points.append(point)
points = np.array(points)
values = f(points.T).T
该数组values
保存每个点的函数值。前一行是唯一f
被调用的地方。下一步,为模型生成单项式,并在这些相同点对它们进行评估。
monomials = [np.zeros((1, n)), np.eye(n)]
for i in range(n):
for j in range(i, n):
monom = np.zeros((1, n))
monom[0, i] += 1
monom[0, j] += 1
monomials.append(monom)
monomials = np.concatenate(monomials, axis=0)
monom_values = np.prod(points**monomials[:, None, :], axis=-1).T
让我们回顾一下这种情况:我们values
这里有函数的形状 (13, 2) 和单项式的形状 (13, 10)。这里 13 是点的数量,10 是单项式的数量。对于 的每一列values
,该lstsq
方法将找到monomials
最接近它的列的线性组合。这些是我们想要的系数。
coeffs = np.linalg.lstsq(monom_values, values, rcond=None)[0]
让我们看看这些是否有好处。系数是
[[1. 1. ]
[1.01171761 3.03011523]
[2.01839762 2.01839762]
[3.03011523 1.01171761]
[0.50041681 4.53385141]
[2.00667556 6.04011017]
[3.02759266 3.02759266]
[2.00667556 2.00667556]
[6.04011017 2.00667556]
[4.53385141 0.50041681]]
和数组monomials
,供参考,是
[[0. 0. 0.]
[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]
[2. 0. 0.]
[1. 1. 0.]
[1. 0. 1.]
[0. 2. 0.]
[0. 1. 1.]
[0. 0. 2.]]
因此,例如,x**2
编码为的单项式 获得函数 的两个分量的[2, 0, 0]
系数。这是完全有道理的,因为它在泰勒展开式中的系数是 0.5,而在它的泰勒展开式中是 4.5。[0.50041681 4.53385141]
f
exp(x + 2*y + 3*z)
exp(3*x + 2*y + z)
函数 f 的近似值可由下式获得
def fFit(point,coeffs,monomials):
return np.prod(point**monomials[:, None, :], axis=-1).T.dot(coeffs)[0]
testpoint = np.array([0.05,-0.05,0.0])
# true value:
print(f(testpoint)) # output: [ 0.95122942 1.0512711 ]
# approximation:
print(fFit(testpoint,coeffs,monomials)) # output: [ 0.95091704 1.05183692]