반응형 mnist1 [머신러닝] Pytorch Linear Regression & MLP - Mnist 활용 이전까지는 원리와 이해를 위한 글을 정리했었는데 이번에는 실제 사용하는 방법과 비슷하게 구현을 해보겠다. 2022.07.05 - [Studying/Machine Learning] - [머신러닝] Linear regression(선형 회귀) 구현 사용 모델은 linear regression과 이를 이용한 MLP다. Linear Regression은 위 포스트를 참고하면 된다. Pytorch에서 모델을 학습시키는 프로세스는 크게 3단계로 나뉜다. 1. 모델에 맞게 데이터 전처리 2. 모델 설계 3. 모델 학습 이 과정을 Mnist dataset(손으로 쓴 숫자 이미지)을 활용해서 진행해보겠다. 우선 패키지를 import 하겠다. import torch import torch.nn as nn import ma.. 2022. 7. 18. 이전 1 다음 반응형