python实现最速下降法求二元函数极值并绘制3D图像

最新推荐文章于 2024-03-02 15:48:54 发布

原创

最新推荐文章于 2024-03-02 15:48:54 发布 · 2.5k 阅读

23 ·

CC 4.0 BY-SA版权

文章标签：

#python

目标

原函数： f(x) = 2x1x1 + x2x2 -2x1*x2 - 4x1 + 4
起始点为（0，0）
精度：误差要求小于 10**（-3）

过程：
在这里插入图片描述

直接上代码看注释吧（）

import numpy as np
import math
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D


def f(x):
    return 2*x[0]**2 + x[1]**2 - 2*x[0]*x[1]-4*x[0]+4   

def step(x):      # 计算每次迭代的步长，H为该函数的hessian矩阵
    H = np.matrix([[4, -