news 2026/7/27 10:52:55

pytorch深度学习笔记

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
pytorch深度学习笔记

目录

摘要

1.神经网络的梯度计算

2.梯度下降法(Gradient Descent)


摘要

本篇文章继续学习尚硅谷深度学习教程,学习内容是神经网络的梯度计算,代码实现用梯度下降法

1.神经网络的梯度计算

在神经网络的学习中,梯度的计算非常重要。神经网络中的梯度,指的就是损失函数关于权重参数的梯度。我们以一个单层的简单网络为例,形状为2×3,权重参数为W,损失函数记为L。那么它的权重参数和梯度为:

这里,梯度也是一个2×3的矩阵,其中各个元素由L关于W中各元素的偏导数构成。

代码如下:

import numpy as np from common.functions import softmax,cross_entropy_error from common.gradient import numerical_gradient class simpleNet: def __init__(self): self.W = np.random.randn(2,3) def forward(self, x): a= x @ self.W return softmax(a) def loss(self, x, t): y = self.forward(x) loss = cross_entropy_error(y, t) return loss x = np.array([0.6, 0.9]) t = np.array([0, 0, 1]) net = simpleNet() f = lambda w: net.loss(x, t) dW = numerical_gradient(f, net.W) print(dW)

测试:

2.梯度下降法(Gradient Descent

梯度下降法(Gradient Descent是一种用于最小化目标函数的迭代优化算法。核心是沿着目标函数(如损失函数)的负梯度方向逐步调整参数,从而逼近函数的最小值。梯度方向指示了函数增长最快的方向,因此负梯度方向是函数下降最快的方向。

具体来说,我们初始找到函数f(x1,x2)的一个点(x1,x2),按下式进行更新

这样就可以沿着负梯度方向,找到一个新的点(x1',x2'),让函数值更小。这里的η表示每次的更新量,在神经网络的学习过程中,就代表了一次学习的步长(一次学习多少、多大程度去更新参数),称为学习率(learning rate)。学习率需要预先设定好,过大或过小都会导致学习效果不佳。

梯度下降法可以代码实现如下:

import numpy as np import matplotlib.pyplot as plt from common.gradient import numerical_gradient def gradient_descent(f, init_x, lr=0.01, step_num=100): x = init_x #保存x变化 x_history = [] for i in range(step_num): x_history.append( x.copy() ) grad = numerical_gradient(f, x) x -= lr * grad return x, np.array(x_history)

案例测试

#定义目标函数f(x)=x1^2+x2^2 def f(x): return x[0]**2 + x[1]**2 if __name__ == '__main__': init_x=np.array([-3.0,4.0]) #超参数 lr=0.1 num_iter=20 #梯度下降法计算最小值 x, x_history = gradient_descent(f, init_x, lr, num_iter) print("最小值点为:",x)

测试

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/26 2:36:21

Langchain-Chatchat Kubernetes集群部署策略

Langchain-Chatchat Kubernetes集群部署策略 在企业对数据隐私和AI合规性要求日益严格的今天,如何构建一个既安全又高效的本地知识库问答系统,成为许多组织面临的核心挑战。公有云上的通用大模型虽然功能强大,但难以满足敏感信息不出内网的需…

作者头像 李华
网站建设 2026/7/27 5:51:39

Langchain-Chatchat日志监控与性能分析最佳实践

Langchain-Chatchat日志监控与性能分析最佳实践 在企业级 AI 应用日益普及的今天,将大型语言模型(LLM)部署于本地环境已成主流趋势。数据安全、低延迟响应和系统可控性成为决策的关键因素。开源项目 Langchain-Chatchat 凭借其对私有知识库的…

作者头像 李华
网站建设 2026/7/26 15:48:42

Langchain-Chatchat模型微调指南:适配垂直领域任务

Langchain-Chatchat模型微调指南:适配垂直领域任务 在企业智能化转型的浪潮中,一个现实而棘手的问题日益凸显:员工每天花大量时间翻找内部文档——HR制度藏在共享盘深处,IT操作手册分散在多个PDF里,财务流程更新了却没…

作者头像 李华
网站建设 2026/7/25 22:55:16

如何配置IPv6静态路由?解决企业网络难题

说到网络配置,尤其是涉及到IPv6的时候,很多人可能会感到头疼。毕竟,这不仅仅是技术问题,更是关乎到整个企业的数据传输效率和安全性。在这样的背景下,如何正确地配置IPv6静态路由成为了一个值得探讨的话题。先来聊聊为…

作者头像 李华
网站建设 2026/7/23 0:53:47

【Linux网络基础】详解 TCP 面向连接 vs UDP 无连接

详解 TCP 面向连接 vs UDP 无连接 本文详细解析计算机网络传输层两个最重要的协议:TCP (Transmission Control Protocol) 和 UDP (User Datagram Protocol),重点阐述“面向连接”与“无连接”的核心区别、工作原理及应用场景。1. 核心概念:什…

作者头像 李华
网站建设 2026/7/26 17:56:23

Langchain-Chatchat如何评估问答质量?指标体系构建

Langchain-Chatchat如何评估问答质量?指标体系构建 在企业知识管理日益智能化的今天,一个常见的困境是:员工每天要花数小时查找文档、邮件或内部系统中的信息,而客服面对客户提问时却常常无法快速调取准确答案。尽管大语言模型&am…

作者头像 李华