This model is implemented as a softmax classifier using PyTorch.
发布时间
阅读量：
阅读量
一、概述
本研究依托pytorch平台，遵循softmax分类器的基本理论，提供该分类器的源代码，以完成图像分类任务。通过代码解析的方式说明神经网络的构建流程，并利用测试集对训练结果进行验证，最终在文末附上完整的代码内容。
二、softmax分类器的搭建
1、获取数据集
import numpy as np
import torch
from torch.utils.data import DataLoader,Dataset
from torchvision import transforms
import gzip
import os
import torchvision
import matplotlib.pyplot as plt
import random
#下载数据集
mnist_train =torchvision.datasets.FashionMNIST(root='~/Datasets/FashionMNIST',train=True, download=True,transform=transforms.ToTensor())
mnist_test =torchvision.datasets.FashionMNIST(r全部评论 (0)
还没有任何评论哟~