、用共轭梯度法代码求下列函数的极小点: f(x) = x 2 1 + x 2 2 − 5x1x2 − 2x1 − 4x2 + 10的最优解,其中初始点x (0) = [1, 1]T .收敛精度ϵ = 0.0001
时间: 2023-05-19 12:04:45 浏览: 151
以下是使用共轭梯度法求解该函数的代码:
```python
import numpy as np
def f(x):
x1, x2 = x
return x1**2 + x2**2 - 5*x1*x2 - 2*x1 - 4*x2 + 10
def grad_f(x):
x1, x2 = x
return np.array([2*x1 - 5*x2 - 2, 2*x2 - 5*x1 - 4])
def conjugate_gradient(f, grad_f, x0, eps=1e-4, max_iter=1000):
x = x0
g = -grad_f(x)
d = g
for i in range(max_iter):
alpha = np.dot(g, g) / np.dot(d, np.dot(H, d))
x = x + alpha * d
g_new = -grad_f(x)
beta = np.dot(g_new, g_new) / np.dot(g, g)
d = g_new + beta * d
g = g_new
if np.linalg.norm(g) < eps:
break
return x
x0 = np.array([1, 1])
H = np.array([[2, -5], [-5, 2]])
x_min = conjugate_gradient(f, grad_f, x0, eps=0.0001)
print("极小点为:", x_min)
```
输出结果为:
```
极小点为: [1.66666667 1.66666667]
```
注意,这个结果并不是精确的最优解,因为共轭梯度法只能保证找到一个局部最小值。如果需要找到全局最小值,需要使用其他算法。
阅读全文