（刘二大人）PyTorch深度学习实践-处理多维输入特征

最新推荐文章于 2023-01-02 00:04:35 发布

Catherine_bling

最新推荐文章于 2023-01-02 00:04:35 发布

阅读量456

点赞数

分类专栏： PyTorch 文章标签：深度学习 pytorch python

本文链接：https://blog.csdn.net/Catherine_bling/article/details/127378219

版权

1.代码实现(带动量，momentum=0.9)

import torch
import numpy as np
from torch.utils.tensorboard import SummaryWriter
import time

#追踪
writer = SummaryWriter("../LEDR")

# 加载数据
xy = np.loadtxt('E:\learn_pytorch\LE\diabetes\diabetes.csv.gz', dtype=np.float32, delimiter=',')
x_data = torch.from_numpy(xy[:,:-1])#全部行，去掉最后一列
y_data = torch.from_numpy(xy[:,[-1]])#这个-1一定要加上中括号，不然取出是向量的形式，我们要求x，y都是矩阵

#构建网络模型
class MultiLogistic(torch.nn.Module):
    def __init__(self):
        super(MultiLogistic, self).__init__()
        self.linear1 = torch.nn.Linear(8,6)
        self.linear2 = torch.nn.Linear(6,4)
        self.linear3 = torch.nn.Linear(4,1)
        self.activation = torch.nn.Sigmoid()

    def forward(self,x):
        x = self.activation(self.linear1(x))
        x = self.activation(self.linear2(x))
        x = self.activation(self.linear3(x))#