ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

python使用梯度下降拟合直线

2022-01-30 11:59:42  阅读:74  来源: 互联网

标签:直线 python 梯度 random epoch 拟合 print wx


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,梯度,random,epoch,拟合,print,wx
来源: https://blog.csdn.net/qq_39303984/article/details/122751512

专注分享技术,共同学习,共同进步。侵权联系[admin#icode9.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有