Pytorch深度学习-梯度下降02

之前采用的穷举法来求解最优的w值,但是当数据量大维度多的时候,会导致训练性能急剧下降
而梯度下降算法的原理:迭代找到目标函数的最小值,或者收敛到最小值

梯度的概念:

1.在单变量的函数中,梯度其实就是函数的微分,代表着函数在某个给定点的切线的斜率
2.在多变量函数中,梯度是一个向量,向量有方向,梯度的方向就指出了函数在给定点的上升最快的方向
3.α在梯度下降算法中被称作为学习率或者步长,意味着我们可以通过α来控制每一步走的距离

随机梯度下降算法:

之前的梯度下降可能会产生鞍点问题,当训练遇到梯度为0的情况时w=w-a*gradient(x_data,y_date),w无法继续向前迭代,有可能会错过最优值,故可以采取随机梯度下降算法,将每一个样本的梯度代入进行迭代,尽可能的将w送到最优处。

本次使用的模型为简单线性回归模型:y = w*x

1.定义线性回归模型

#定义模型
def forward(x):
    return w*x
#初始随机猜测一个权值w=1.0
w=-1.0

2.计算损失函数

#定义单次成本计算函数
def loss(x,y):
    y_pred = forward(x)
    return (y-y_pred) ** 2

3.数据集

#定义 DataSet-实际w==2
x_date = [1.0,2.0,3.0]
y_date = [2.0,4.0,6.0]

4.定义梯度计算公式

#定义随机梯度计算公式
def gradient(x, y):
    return 2*x*(x*w-y)

5.使用梯度下降算法进行训练

#通过梯度下降算法来训练模型---设定为1-100次:左闭右开区间
#拿所有样本对反复训练100次,每一次都需要训练所有的样本对---梯度为样本点的梯度
for epoch in range(1, 100):
    print("epoch=", epoch)
    for x, y in zip(x_date, y_date):
        loss_val = loss(x, y)
        # 每一次训练用的梯度都是一对样本对的梯度,也就是某点的梯度
        w = w-0.01*gradient(x, y)   #学习率定义为0.01
        print("\tx=", x, "y=", y, "grad=", gradient(x, y),"w=", w)
    #打印输出是第几次训练,训练权值和训练的损失函数
    w_list.append(w)
    mse_list.append(loss_val)
    print("\tw=", w, "loss=", loss_val)

6.完整代码及训练结果

# -*- codeing = utf-8 -*-
'''
梯度下降算法存在的问题:
    1.陷入局部最优
    2.解决鞍点问题
'''
#采用随机梯度下降算法---计算每一点样本对的梯度来进行w权值的更新
#优点: 时间复杂度高,但是结果比较准确
#存在的缺点: 由于本次的训练w值来源于上一个样本的梯度训练,故由于前后关系,无法进行并行计算的方式进行梯度下降训练
'''
当计算平均梯度遇到鞍点(梯度为0点,即gradient == 0)时,w=w-a*gradient无法前进,此时某一随机样本的梯度可能不是鞍点,可能继续将w向前
'''
import matplotlib.pyplot as plt

#训练数据集---实际w权值=2.0
x_date = [1.0,2.0,3.0]
y_date = [2.0,4.0,6.0]
#初始随机猜测一个权值w=1.0
w=-1.0
#定义线性回归模型
def forward(x):
    return x*w
#定义单次成本计算函数
def loss(x,y):
    y_pred = forward(x)
    return (y-y_pred) ** 2
w_list = []
mse_list = []
#定义随机梯度计算公式
def gradient(x, y):
    return 2*x*(x*w-y)
print("predict before training:", "w=", w, forward(4))

#通过梯度下降算法来训练模型---设定为1-100次:左闭右开区间
#拿所有样本对反复训练100次,每一次都需要训练所有的样本对---梯度为样本点的梯度
for epoch in range(1, 100):
    print("epoch=", epoch)
    for x, y in zip(x_date, y_date):
        loss_val = loss(x, y)
        # 每一次训练用的梯度都是一对样本对的梯度,也就是某点的梯度
        w = w-0.01*gradient(x, y)   #学习率定义为0.01
        print("\tx=", x, "y=", y, "grad=", gradient(x, y),"w=", w)
    #打印输出是第几次训练,训练权值和训练的损失函数
    w_list.append(w)
    mse_list.append(loss_val)
    print("\tw=", w, "loss=", loss_val)

print("predict completed:", "w=", w, forward(4))
plt.plot(w_list, mse_list)
plt.ylabel("mean Square error")
plt.xlabel("w")
plt.show()



结果如下:
image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,033评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,725评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,473评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,846评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,848评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,691评论 1 282
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,053评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,700评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,856评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,676评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,787评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,430评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,034评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,990评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,218评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,174评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,526评论 2 343

推荐阅读更多精彩内容