Pytorch로 시작하는 딥러닝 입문(06-05. XOR: 다중 퍼셉트론 구현하기)

♣  파이토치로 다중 퍼셉트론 구현하기

 

 

입력과 출력 정의하기

 

 

다층 퍼셉트론 설계하기

다층 퍼셉트론을 설계하는데, 아래는 입력층, 은닉층1, 은닉층2, 은닉층3, 출력층을 가지는 인공신경망이다. 

 

위 인공신경망을 그림으로 표현하면 아래와 같다. 

 

비용함수와 옵티마이저 선언하기

nn.BCELoss( )는 이진 분류에서 사용하는 크로스엔트로피 함수이다. 

 

학습을 진행하면서 비용이 최소화되는 방향으로 가중치와 편향이 업데이트된다. 

 

 

♣ 학습된 다층 퍼셉트론의 예측값 확인하기

모델이 XOR 문제를 풀 수 있는지 테스트해보자. 

 

실제값은 0, 1, 1, 0이며 예측값은 0, 1, 1, 0으로 문제를 해결할 수 있다는 것을 보여준다!