pytorch入门

1.数据结构

        使用pytorch需要将数据转换成模型需要的数据格式,数据格式包括单个数据对应的scalar格式,表示特征的vector格式,对应矩阵的matrix格式,对应多维数据的n-ddimensional tensor格式。数据转化有以下的方式。

'''使用tnesor进行数据转换'''
import torch
torch.tensor(input_features[start:end]


'''使用TensorDataset,DataLoader 进行数据格式修改'''
from torch.utils.data import TensorDataset
from torch.utils.data import DataLoader

bs = 64                                                               #数据分组大小
train_ds = TensorDataset(x_train , y_train)                           #将数据封装成tensor的格式
train_dl = DataLoader(train_ds,batch_size=bs ,shuffle=True)           #将数据安装batch_size的方式分割

for xb,yb in train_dl:                                                #这里可以直接调用
    loss_batch(model,loss_func,xb,yb,opt)

2.主要的包

1. torch.nn.Module  

        在建立模型的基本结构时使用,主要在建立模型有参数的结构时使用,比如建立模型的全链接层和池化层的时候使用,同时在定义模型的基本结构时作为继承的父类使用。

2. torch.nn.functional 

        在建立模型的基本结构时使用,主要在建立模型没有参数的结构时使用,比如建立模型的激活函数,损失函数时候使用

class Mnist_NN(nn.Module):                       # 建立模型的基本结构时需要
    def __init__(self):
        super().__init__()
        self.hidden1 = nn.Linear(784,128)        # 建立全链接层时需要
        self.hidden2 = nn.Linear(128,256)
        self.out = nn.Linear(256,10)

    def forward(self,x):
        x = F.relu(self.hidden1(x))              #建立激活函数时需要
        x = F.relu(self.hidden2(x))
        x = self.out(x)
        return x

3 以较为复杂方式实现回归

import pandas as pd
import numpy as np
import torch
from sklearn import preprocessin
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值