展示MNIST数据集

﹏ヽ暗。殇╰゛Y 2022-04-05 09:43 364阅读 0赞

import torch
    import torch.nn as nn
    import torchvision
    import torchvision.transforms as transforms
    import matplotlib.pyplot as plt
    
    # Device configuration
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    
    # Hyper-parameters
    input_size = 784
    hidden_size = 500
    num_classes = 10
    num_epochs = 5
    batch_size = 100
    learning_rate = 0.001
    
    # MNIST dataset
    train_dataset = torchvision.datasets.MNIST(root='../../data',
                                               train=True,
                                               transform=transforms.ToTensor(),
                                               download=True)
    
    test_dataset = torchvision.datasets.MNIST(root='../../data',
                                              train=False,
                                              transform=transforms.ToTensor())
    
    # Data loader
    train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
                                               batch_size=batch_size,
                                               shuffle=True)
    
    test_loader = torch.utils.data.DataLoader(dataset=test_dataset,
                                              batch_size=batch_size,
                                              shuffle=False)
    print(train_loader)
    
    
    fig = plt.figure()
    for i in range(1,20):
          plt.imshow(train_dataset.train_data[i].numpy(), cmap='gray')
          plt.title('%i' % train_dataset.train_labels[i])
    
          plt.show()

发表评论取消回复

表情：

评论列表（有 0 条评论，364人围观）

还没有评论，来说两句吧...

相关阅读

相关 Fashion-MNIST数据集入门

目录 Fashion-MNIST数据集入门加载Fashion-MNIST数据集数据预处理构建模型编译和训练模型评估模型总结 1. 图像分类 2.

旧城等待，/ 2024年02月21日 12:16/ 0 赞/ 20 阅读

相关详解 MNIST 数据集

MNIST 数据集已经是一个被”嚼烂”了的数据集, 很多教程都会对它”下手”, 几乎成为一个 “典范”. 不过有些人可能对它还不是很了解, 下面来介绍一下. MNIST 数据

怼烎@/ 2022年06月12日 06:15/ 0 赞/ 361 阅读

相关 C语言解析MNIST数据集

小编用C语言解析MNIST数据集，产生的图片是倒立着的。数据集下载地址：[http://yann.lecun.com/exdb/mnist/][http_yann.le

布满荆棘的人生/ 2022年06月12日 02:07/ 0 赞/ 257 阅读

相关模仿mnist数据集制作自己的数据集

模仿mnist数据集制作自己的数据集最近看深度学习，用TensorFlow跑教程里的模型，发现教程里大多都是用mnist和cifar这两个数据集来演示的。想测

曾经终败给现在/ 2022年06月10日 00:08/ 0 赞/ 451 阅读

相关展示MNIST数据集

import torch import torch.nn as nn import torchvision import torchvision

﹏ヽ暗。殇╰゛Y/ 2022年04月05日 09:43/ 0 赞/ 365 阅读

相关 MNIST数据集介绍

MNIST数据集的组成 MNIST数据集包含了6w张图片作为训练数据，1w图片作为测试数据。在MNIST数据集中，每一张图片都代表了0~9中的一个数字，图片的大小都是28

约定不等于承诺〃/ 2022年03月16日 06:52/ 0 赞/ 744 阅读

相关 tensorflow分类任务MNIST数据集

import tensorflow as tf import numpy as np from tensorflow.examples.tutorial

分手后的思念是犯贱/ 2022年02月22日 13:38/ 0 赞/ 348 阅读

相关 MNIST数据集介绍及读取

转载自：《[MINIST数据集介绍及读取][MINIST]》 MNIST数据集是机器学习领域中非常经典的一个数据集，由60000个训练样本和10000个测试样本组成，每

冷不防/ 2022年01月20日 15:17/ 0 赞/ 389 阅读

相关 mnist数据集下载——mnist数据集提供百度网盘下载地址

mnist数据集是由深度学习大神 LeCun等人制作完成的数据集，mnist数据集也常认为是深度学习的“ Hello World！”。官网：[http://yann.lec

女爷i/ 2021年12月14日 12:09/ 0 赞/ 578 阅读

相关 Pytorch 之 MNIST 数据集实现

目录 1. 数据集介绍 2. 代码 2. 读代码（个人喜欢的顺序） 2.1. 导入模块部分： 2.2. Main 函数：

逃离我推掉我的手/ 2021年12月13日 04:02/ 0 赞/ 380 阅读