【pytorch】全连接网络简单二次函数拟合

news2024/12/25 12:58:22

下面是一个使用PyTorch实现全连接网络来拟合简单二次函数 y = x 2 y = x^2 y=x2 的示例。我们将创建一个简单的神经网络,定义损失函数和优化器,并进行训练。
在这里插入图片描述

下面是完整的代码示例:

import torch
import torch.nn as nn
import torch.optim as optim
import matplotlib.pyplot as plt

# 生成数据
x = torch.linspace(-10, 10, 100).unsqueeze(1)
y = x**2

# 定义全连接网络
class SimpleNN(nn.Module):
    def __init__(self):
        super(SimpleNN, self).__init__()
        self.fc1 = nn.Linear(1, 128)
        self.fc2 = nn.Linear(128, 64)
        self.fc3 = nn.Linear(64, 1)

    def forward(self, x):
        x = torch.relu(self.fc1(x))
        x = torch.relu(self.fc2(x))
        x = self.fc3(x)
        return x

# 实例化模型、损失函数和优化器
model = SimpleNN()
criterion = nn.MSELoss()
optimizer = optim.Adam(model.parameters(), lr=0.01)

# 训练模型
num_epochs = 1000
for epoch in range(num_epochs):
    model.train()
    optimizer.zero_grad()
    outputs = model(x)
    loss = criterion(outputs, y)
    loss.backward()
    optimizer.step()

    if (epoch+1) % 100 == 0:
        print(f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item():.4f}')

# 预测
model.eval()
predicted = model(x).detach()

# 可视化结果
plt.plot(x.numpy(), y.numpy(), 'ro', label='Original data')
plt.plot(x.numpy(), predicted.numpy(), 'b-', label='Fitted line')
plt.legend()
plt.show()

这段代码的具体步骤如下:

  1. 生成数据:创建输入数据 ( x ) 和对应的标签 ( y )。
  2. 定义网络结构:创建一个简单的全连接神经网络,包括三层线性层。
  3. 实例化模型、损失函数和优化器:使用均方误差损失函数和Adam优化器。
  4. 训练模型:在1000个epoch上训练模型,并在每100个epoch打印一次损失。
  5. 预测和可视化:使用训练好的模型进行预测,并将原始数据和拟合结果进行可视化。

运行这段代码后,你将看到一个图形,其中红点表示原始的二次函数数据,蓝线表示神经网络拟合的结果。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1983661.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

Linux笔记(1)

在敲出 文件 / 目录 / 命令 的前几个字母之后,按下 tab 键 如果输入的没有歧义,系统会自动补全 如果还存在其他 文件 / 目录 / 命令 ,再按一下 tab 键,系统会提示可能存在的命令 小…

【算法速刷(5/100)】LeetCode —— 20.有效的括号

题目要求比较明晰简洁&#xff0c;编码难度并不算高 下面贴出代码和思路 bool isValid(string s) {stack<char> stk;for(const char& c : s){if(stk.empty()){stk.push(c);continue;}if(c ( || c [ || c {){stk.push(c);continue;}else{char top stk.top();boo…

Java Lambda表达式总结(快速上手图解)

Java Lambda表达式总结&#xff08;快速上手详解&#xff09;-CSDN博客https://blog.csdn.net/m0_66070037/article/details/140912566?spm1001.2014.3001.5501

idea 设置自动移除多余 的import (或整个项目删除)

1、开发的时候&#xff0c;直接自动去除多余的引入 2、如果想把整个项目的移除可以如下操作

Linux 异常 sh: /usr/bin/xauth: not found

异常原因&#xff1a; 远程连接时出现错误&#xff0c;由于 Omega 上没有 X 环境&#xff0c;因此在尝试交换凭据以设置转发时会失败 解决办法&#xff1a; 进入高级 SHH 设置&#xff0c;并勾选了 X11 转发。取消勾选该框。

Studying-代码随想录训练营day59| dijkstra(堆优化版)精讲、Bellman_ford 算法精讲

第59天&#xff0c;dijkstra算法的优化版本&#xff0c;以及Bellman_ford 算法&#x1f4aa;&#x1f4aa;(ง •_•)ง&#xff0c;编程语言&#xff1a;C 目录 dijkstra&#xff08;堆优化版&#xff09;精讲 思路 图的存储 邻接矩阵 邻接表 本题图的存储 堆优化细节…

x86 8086 CPU 寄存器详解

8086 寄存器及其地址详细介绍 8086 微处理器是 Intel 在 1978 年推出的一款 16 位微处理器&#xff0c;它在计算机体系结构的发展中占有重要地位。8086 处理器拥有 14 个 16 位寄存器&#xff0c;这些寄存器被分为几类&#xff1a;通用寄存器、段寄存器、指令指针寄存器和标志…

力扣第五十四题——螺旋矩阵

内容介绍 给你一个 m 行 n 列的矩阵 matrix &#xff0c;请按照 顺时针螺旋顺序 &#xff0c;返回矩阵中的所有元素。 示例 1&#xff1a; 输入&#xff1a;matrix [[1,2,3],[4,5,6],[7,8,9]] 输出&#xff1a;[1,2,3,6,9,8,7,4,5]示例 2&#xff1a; 输入&#xff1a;matrix …

【好用的个人工具】部署Dokcer容器速查表工具

【好用的个人工具】部署Dokcer容器速查表工具 一、getting-started介绍1.1 getting-started简介1.2 getting-started内容 二、本地环境介绍2.1 本地环境规划2.2 本次实践介绍 三、本地环境检查3.1 检查Docker服务状态3.2 检查Docker版本3.3 检查docker compose 版本 四、下载ge…

文心智能体平台:食尚小助,提供美食推荐和烹饪指导

文章目录 前言文心智能体平台介绍创建自己的智能体我的文心智能体体验地址总结 前言 在快节奏的现代生活中&#xff0c;许多人都希望能够享受美味的食物&#xff0c;但往往缺乏时间和精力来自己动手烹饪。为了解决这一问题&#xff0c;文心智能体平台推出了“食尚小助”智能体…

计算机网络基础之网络套接字socket编程(初步认识UDP、TCP协议)

绪论​ “宿命论是那些缺乏意志力的弱者的借口。 ——罗曼&#xff0e;罗兰”&#xff0c;本章是为应用层打基础&#xff0c;因为在写应用层时将直接通过文本和代码的形式来更加可视化的理解网络&#xff0c;本章主要写的是如何使用网络套接字和udp、tcp初步认识。 话不多说安…

“论负载均衡技术在Web系统中的应用”写作框架软考高级论文系统架构设计师论文

论文真题 负载均衡技术是提升Web系统性能的重要方法。利用负载均衡技术&#xff0c; 可将负载(工作任务) 进行平衡、分摊到多个操作单元上执行&#xff0c; 从而协同完成工作任务&#xff0c; 达到提升Web系统性能的目的。 请围绕“负载均衡技术在Web系统中的应用”论题&…

云原生真机实验

基于Proxmox VE构建中小企业云计算平台 首先Proxmox VE是什么&#xff1f;能用来做什么&#xff1f; Proxmox VE是一个完整的企业虚拟化开源平台。借助内置的 Web 界面&#xff0c;可以在单个解决方案上轻松管理 VM(开虚拟机的) 和容器、软件定义的存储和网络、高可用性群集以…

使用Python连接华为云物联网服务器与服务器完成数据交互

一、前言 随着物联网技术的快速发展&#xff0c;越来越多的设备和系统需要通过网络进行连接和数据交换&#xff0c;以实现智能化管理和控制。华为云物联网平台作为业界领先的物联网解决方案提供商&#xff0c;提供了稳定可靠的MQTT服务器&#xff0c;使得设备能够轻松接入云端…

数据结构(其四)--特殊矩阵的存储

目录 11.特殊矩阵的压缩存储 &#xff08;1&#xff09;.一维数组的储存结构 &#xff08;2&#xff09;.二维数组的存储结构 &#xff08;3&#xff09;.普通矩阵的存储 &#xff08;4&#xff09;.特殊矩阵的压缩存储 i.对称矩阵 ii.三角矩阵 iii.三对角矩阵 iiii.稀疏矩…

上位机 OPC协议、KepServerEX

OPC 全称 OLE For Process Control 》》OPC&#xff08;Open Platform Communications&#xff0c;以前称为 OLE for Process Control&#xff09;是一组软件技术 opc出现之前&#xff0c;软件和硬件是分开的&#xff0c; 如果要与不同的设备通信&#xff0c;需要用各个厂商的…

DepthB2R【附代码】(权限提升)

靶机下载地址&#xff1a; https://www.vulnhub.com/entry/depth-1,213/https://www.vulnhub.com/entry/depth-1,213/ 1. 主机发现端口扫描目录扫描 1.1. 主机发现 nmap -sn 192.168.43.0/24|grep -B 2 08:00:27:08:B4:07 1.2. 端口扫描 nmap -p- 192.168.43.112 1.3. 目录…

NoSQL 数据库之MongoDB

MongoDB 是一个开源的 NoSQL 数据库&#xff0c;由 MongoDB Inc. 研发和维护。它采用文档存储模型&#xff0c;使用 JSON 类似的 BSON&#xff08;二进制 JSON&#xff09;格式来存储数据。MongoDB 具有高性能、易扩展和高可用性等特点&#xff0c;广泛应用于现代 web 应用程序…

Linux学习笔记:iptables命令管理

1、iptables简介 其实iptables只是Linux防火墙的管理工具而已&#xff0c;位于/sbin/iptables。真正实现防火墙功能的是netfilter&#xff0c;它是Linux内核中实现包过滤的内部结构。 语法格式&#xff1a;iptables [-t table] COMMAND [chain] CRETIRIA -j ACTION -t&#…