当前位置: 首页 > news >正文

网站建设工作是干什么的江门外贸网站建设

网站建设工作是干什么的,江门外贸网站建设,建设网站跟服务器得关系,宿迁网站建设开发文章目录 1. 创建数据集1.1. 直接继承Dataset类1.2. 使用TensorDataset类 2. 加载数据集3. 将数据转移到GPU 1. 创建数据集 主要是将数据集读入内存,并用Dataset类封装。 1.1. 直接继承Dataset类 必须要重写__getitem__方法,用于根据索引获得相应样本…

文章目录

  • 1. 创建数据集
    • 1.1. 直接继承Dataset类
    • 1.2. 使用TensorDataset类
  • 2. 加载数据集
  • 3. 将数据转移到GPU

1. 创建数据集

主要是将数据集读入内存,并用Dataset类封装。

1.1. 直接继承Dataset类

必须要重写__getitem__方法,用于根据索引获得相应样本数据。必要时还可以重写__len__方法,用于返回数据集的大小。

from torch.utils.data import Datasetclass BostonHousingDataset(Dataset):"""定义波士顿房价数据集"""def __init__(self):self.data = np.load('../dataset/boston_housing/boston_housing.npz')def __getitem__(self, index):return self.data['x'][index], self.data['y'][index]def __len__(self):return self.data['x'].shape[0]

1.2. 使用TensorDataset类

将多个张量组合成一个数据集,要保证所有张量的第一个维度相等,保证每批样本数据格式相同。

import torch
from torch.utils.data import TensorDatasetdata = np.load('../dataset/boston_housing/boston_housing.npz')
X = torch.tensor(data['x'])
y = torch.tensor(data['y'])
dataset = TensorDataset(X, y)

2. 加载数据集

使用DataLoader类将Dataset封装的数据集分成批次并进行迭代,以便于模型训练。DataLoader常用参数如下:

  • dataset
    要加载的数据集。
  • batch_size
    每个数据批次中包含的样本数。默认为1。
  • shuffle
    是否打乱数据集。默认为False。
  • num_workers
    使用几个进程来加载数据。默认为0,即在主进程中加载数据。
  • drop_last
    当数据集样本数不能被batch_size整除时,是否舍弃最后一个不完整的batch。默认为False。
from torch.utils.data import DataLoaderdataloader = DataLoader(dataset, batch_size=16, shuffle=True)

3. 将数据转移到GPU

一般在要运算时才将数据转移到GPU,有以下两种方法:

  1. var.to(device)
  2. var.cuda()
import torchdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")
for X,y in dataloader:# 将数据转移到GPUX = X.to(device)y = y.to(device)# 也可以X = X.cuda()y = y.cuda()
http://www.yayakq.cn/news/774401/

相关文章:

  • 彩票网站代理怎么做网站建设要做什么
  • 兰州新区规划建设管理局网站网站建设找盛誉网络
  • 网站与系统对接图文方案大鹏网站建设建站好不好
  • 中国网站建设集团如何做网站水晶头
  • 高端人才招聘网站排名域名查询万网
  • 学会网站制作要多久莆田手表网站
  • 网站上传不了wordpress 支付宝付款
  • 网站积分的作用素材羊设计师服务平台
  • 教育网站建设情况报告温州制作网站软件
  • 网站与网址的区别查询网站名有没有收录
  • zencart 网站安装如何用付费音乐做视频网站
  • 网站风格分析微信公众号怎么做商城
  • 龙岗网站制作设计网页设计师培训排行榜
  • 苏州建站公司哪个好如东城乡建设局网站
  • 中国佛山营销网站建设电子商务网站建设案例分析
  • 替老外做网站中国建设网官方网站建筑工程税率
  • 高密微网站建设百度竞价
  • 北京学设计去哪个网站高端制造股十大龙头
  • 网站备案知识交换友情链接的渠道有哪些
  • 哪些网站做装修闽清住房和城乡建设局网站
  • 家用电脑当服务器建设网站河北石家庄天气
  • 最专业的网站开发公司哪家最专业山东省住房建设部网站首页
  • 物流那个网站做推广好网站程序合同
  • 江阴建设网站西安工程信息交易中心
  • 网站模板 asp pc wapwordpress采集英文
  • 网站建设建设意见wordpress二次元美化
  • 建筑网站设计在线代理网页版proxy
  • 免费php网站模板wordpress 跳转 文章
  • 用jsp做校园网站专业做包装的电商网站
  • 网站建设开发公司重庆建筑安全信息网