pytorch 找最大值

最新推荐文章于 2024-02-14 22:57:12 发布

大话幺儿

最新推荐文章于 2024-02-14 22:57:12 发布

阅读量6.1k

点赞数 1

分类专栏： pytorch 文章标签：深度学习神经网络 pytorch

本文链接：https://blog.csdn.net/dahuayaoer/article/details/104797770

版权

本文介绍了如何使用PyTorch构建神经网络从数组中寻找最大值。针对训练集中样本类别不平衡（大部分为最大值）的问题，文章讨论了全连接网络+分类方法的不足，并给出了具体的数据集构造示例。

摘要由CSDN通过智能技术生成

题意：使用神经网络实现，从数组中找出最大值。

提供数据：两个 csv 文件，一个存训练集：n 个 m 维特征自然数数据，另一个存每条数据对应的 label ，就是每条数据中的最大值。这里将随机构建训练集：

#%%
import numpy as np 
import pandas as pd 
import torch 
import random 
import torch.utils.data as Data
import torch.nn as nn
import torch.optim as optim


def GetData(m, n):
    dataset = []
    for j in range(m):
        max_v = random.randint(0, 9)
        data = [random.randint(0, 9) for i in range(n)]
        dataset.append(data)
    label = [max(dataset[i]) for i in  range(len(dataset))]
    data_list = np.column_stack((dataset, label))
    data_list = data_list.astype(np.float32)
    return data_list

#%%
# 数据集封装 重载函数len, getitem
class GetMaxEle(Data.Dataset):
    def __init__(self, trainset):
        self.data = trainset 

    def __getitem__(self, index):
        item = self.data[index]
        x = item[:-1]
        y = i