728x90
반응형

pytorch mlp 2

[PyTorch] MLP 모델 만들기 2편 - 학습 모델 구축하기

지난 1편에서는 학습 데이터셋 만들어 보았습니다.https://notesfromyoungdeveloper.tistory.com/25 [PyTorch] MLP 모델 만들기 1편 - 학습 데이터셋 만들기PyTorch에서 MLP 모델을 만드는 방법에 대해 알아보겠습니다.MLP는 Multi-Layer Perceptron의 약자로, 입력층(input layer)과 출력층(output layer) 사이에 1개 이상의 은닉층(hidden layer)이 있는 구조의 퍼셉트론을notesfromyoungdeveloper.tistory.com 2편에서는 MLP 모델 구축하는 방법에 대해 알아보겠습니다.먼저, 전체 코드는 아래와 같고, 하나씩 살펴보겠습니다.class XORModel(pl.LightningModule): ..

[PyTorch] MLP 모델 만들기 1편 - 학습 데이터셋 만들기

PyTorch에서 MLP 모델을 만드는 방법에 대해 알아보겠습니다.MLP는 Multi-Layer Perceptron의 약자로, 입력층(input layer)과 출력층(output layer) 사이에 1개 이상의 은닉층(hidden layer)이 있는 구조의 퍼셉트론을 말합니다. 만들어볼 MLP 모델은 XOR 게이트 모델입니다.이번 글에서는 모델을 만드는 과정 중 학습 데이터셋 만드는 부분을 다뤄보겠습니다. 먼저, 입력 데이터의 리스트와 출력 데이터의 리스트를 만들어 줍니다.그리고 zip 함수를 이용해서 2개의 리스트를 하나로 묶어줍니다. 그럼 이어서 실제 코드를 살펴보겠습니다.import torchfrom torch.autograd import Variablexor_input = [Variable(tor..

728x90
반응형