python实现梯度下降

最新推荐文章于 2024-07-08 11:43:01 发布

墙角的秋天

最新推荐文章于 2024-07-08 11:43:01 发布

阅读量1.4k

点赞数

分类专栏：数据挖掘算法文章标签：梯度下降

本文链接：https://blog.csdn.net/wangxingfan316/article/details/82054712

版权

数据挖掘算法专栏收录该内容

6 篇文章 0 订阅

订阅专栏

1、Rosenbrock函数

#！／user/bin/env python
#-*- coding:utf-8 -*-
#梯度下降Rosenbrock函数

def rb(x,y):
    #定义rosenbrock函数
    return (1-x)**2+100*(y-x**2)**2

def partial_x(x,y):
    #计算x的偏导数
    return -2*(1-x)-400*x*(y-x**2)

def partial_y(x,y):
    #计算y的偏导数
    return 200*(y-x**2)

def bgd_rosenbrock():
    x=0
    y=0
    i=0
    learning_rate = 0.001
    while i < 10000:
    #步长0.001下降10000步
        x = x - learning_rate * partial_x(x, y)
        y = y - learning_rate * partial_y(x, y)
        print('min',rb(x,y))
        i = i+1
bgd_rosenbrock()