python使用梯度下降拟合直线

本文主要是介绍python使用梯度下降拟合直线，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

python使用梯度下降拟合直线

#目标直线y=w_t*x+b_t
import random
#定义学习率
a=0.01
epoch = 3000
#初始化
w=random.random()
b=random.random()
w_t=random.random()
b_t=random.random()
print(w_t,w,b_t,b)
for i in range(epoch):
	#生成样本数据
    x=random.random()
    y=w_t*x+b_t
    #函数(y-wx-b)*(y-wx-b)的偏导数
    g_w=-2*(y-w*x-b)*x
    g_b=-2*(y-w*x-b)
    #更新
    w=w-a*g_w
    b=b-a*g_b
print(w,b)

运行结束时w,b和w_t,b_t的值基本接近

这篇关于python使用梯度下降拟合直线的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！