반응형 파이썬 머신러닝 예제/MNIST 분류예제 (딥러닝)1 [파이토치] MNIST 분류예제 딥러닝 파이토치를 이용하여 MNIST 를 분류하는 예제입니다. 자세한 설명을 주석에 적어놓았습니다. import matplotlib.pyplot as plt from torchvision.datasets.mnist import MNIST from torchvision.transforms import ToTensor from torch.utils.data.dataloader import DataLoader import torch import torch.nn as nn from torch.optim.adam import Adam #MNIST 손글씨 데이터를 불러옵니다. transform 을 통해 텐서 데이터로 바꿔줍니다. #train 과 test 데이터 중 어떤 데이터를 불러올지는 train 옵션으로 설정합니다. .. 2023. 10. 27. 이전 1 다음 반응형