news 2026/4/16 12:36:12

使用pytorch进行batch_size分批训练,并使用adam+lbfgs算法——波士顿房价预测

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
使用pytorch进行batch_size分批训练,并使用adam+lbfgs算法——波士顿房价预测

使用pytorch进行batch_size分批训练,并使用adam+lbfgs算法

  • 数据探索
  • 训练过程及结果
  • 整批次训练与分批次训练对比
  • 绘制结果对比曲线
  • 绘制无序曲线对比结果图

使用pytorch神经网络进行波士顿房价预测

数据探索

训练过程及结果


importnumpyasnpimportpandasaspdimportmatplotlib.pyplotaspltfromsklearn.model_selectionimporttrain_test_splitfromsklearn.preprocessingimportStandardScalerimporttorchimporttorch.nnasnnimporttorch.optimasoptimfromtqdmimporttqdm url="https://raw.githubusercontent.com/Zhang-bingrui/Boston_house/refs/heads/main/house_data.csv"boston_df=pd.read_csv(url,header=0,on_bad_lines="skip"# 跳过格式错误的行,防止报错)X=boston_df.drop('MEDV',axis=1).values y=boston_df['MEDV'].values#划分训练集和测试集# Veriyi %20 test setine ve %80 eğitim setine bölelimX_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.3,random_state=42)#输入数据标准化scaler=StandardScaler()X_train_scaled=scaler.fit_transform(X_train)X_test_scaled=scaler.transform(X_test)#将数据转换为pytorch的TENSORX_train=torch.tensor(X_train_scaled,dtype=torch.float32)X_test=torch.tensor(X_test_scaled,dtype=torch.float32)y_train=torch.tensor(y_train,dtype=torch.float32).view(-1,1)y_test=torch.tensor(y_test,dtype=torch.float32).view(-1,1)#创建数据加载器train_dataset=TensorDataset(X_train,y_train)test_dataset=TensorDataset(X_test,y_test)train_loader=DataLoader(train_dataset,batch_size=64,shuffle=True)test_loader=DataLoader(test_dataset,batch_size=64,shuffle=False)# ANN modellerini tanımlayalımclassANN(nn.Module):def__init__(self,input_dim):super(ANN,self).__init__()self.fc1=nn.Linear(input_dim,64)self.fc2=nn.Linear(64,32)self.fc3=nn.Linear(32,1)defforward(self,x):x=torch.relu(self.fc1(x))x=torch.relu(self.fc2(x))x=self.fc3(x)returnx num_epochs=500switch_epoch=
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/16 10:16:10

LobeChat医疗健康问答合规性讨论

LobeChat 在医疗健康问答中的合规性实践与架构设计 在智慧医疗快速发展的今天,越来越多的医院和健康管理机构开始尝试引入人工智能技术来提升服务效率。然而,当大语言模型(LLM)被用于回答“我最近头晕是不是脑梗前兆?”…

作者头像 李华
网站建设 2026/4/16 10:14:31

Linux 线程

一、线程核心概念 1. 定义与归属 线程是轻量级进程(Lightweight Process, LWP),必须隶属于某个进程,不能独立存在。进程是线程的 "容器",线程的所有操作都依赖于进程的资源(如代码段、数据段、…

作者头像 李华
网站建设 2026/4/16 10:13:58

Slick轮播组件样式架构深度解析与定制实践

Slick轮播组件样式架构深度解析与定制实践 【免费下载链接】slick the last carousel youll ever need 项目地址: https://gitcode.com/GitHub_Trending/sl/slick 在现代前端开发中,轮播组件作为用户界面交互的核心元素,其样式定制能力直接影响产…

作者头像 李华
网站建设 2026/4/16 9:01:23

NVIDIA Profile Inspector终极指南:5分钟掌握专业级显卡优化技巧

NVIDIA Profile Inspector终极指南:5分钟掌握专业级显卡优化技巧 【免费下载链接】nvidiaProfileInspector 项目地址: https://gitcode.com/gh_mirrors/nv/nvidiaProfileInspector 想要彻底释放你的NVIDIA显卡性能吗?NVIDIA Profile Inspector作…

作者头像 李华
网站建设 2026/4/16 12:22:10

比手动排查快10倍:自动化修复Python库缺失问题

快速体验 打开 InsCode(快马)平台 https://www.inscode.net输入框内输入如下内容: 开发一个自动化诊断工具,输入错误信息importerror: libpython3.8.so.1.0后自动执行:1) 系统环境检测;2) Python安装验证;3) 依赖关系…

作者头像 李华
网站建设 2026/4/16 10:59:44

Qt打包入门:5分钟学会生成你的第一个exe

快速体验 打开 InsCode(快马)平台 https://www.inscode.net输入框内输入如下内容: 开发一个极简的Qt打包教学工具,要求:1. 提供step-by-step向导界面;2. 自动检测系统Qt环境;3. 内置简单的Qt示例项目;4. 可…

作者头像 李华