목록소프트맥스 (1)

윤제로의 제로베이스

소프트맥스 회귀로 MNIST 데이터 분류하기

https://wikidocs.net/60324 05. 소프트맥스 회귀로 MNIST 데이터 분류하기 이번 챕터에서는 MNIST 데이터에 대해서 이해하고, 파이토치(PyTorch)로 소프트맥스 회귀를 구현하여 MNIST 데이터를 분류하는 실습을 진행해봅시다. MNIST ... wikidocs.net for X, Y in data_loader: # 입력 이미지를 [batch_size × 784]의 크기로 reshape # 레이블은 원-핫 인코딩 X = X.view(-1, 28*28) import torch import torchvision.datasets as dsets import torchvision.transforms as transforms from torch.utils.data import Data..
Background/Pytorch 기초 2022. 1. 16. 21:59
이전 Prev 1 Next 다음