[Pytorch] Class를 이용한 Model 구현

https://wikidocs.net/60754

유원준 님의 Pytorch로 시작하는 딥러닝 입문

 

이전 포스팅

https://yscho.tistory.com/44

 

대부분의 Pytorch 상의 모델은 클래스로 구현한다. 이번에는 클래스를 통한 구현하는 예제를 익히고자 한다. 

 

앞선 포스팅에서 nn.Linear를 이용하여 모형을 구현한 예제를 살펴보았다. 

이는 클래스로 만들면 더 편하게 사용할 수 있는데 다음과 같이 사용하는 것이 일반적이다. 

 

 

다음과 같이 객체로 정의하는 것이다. 

 

그렇다면 클래스 객체 안의 내용을 자세하게 파헤쳐보자

 

우선 모형을 만들 때 nn.Module을 상속받고 super() 함수를 통해 자동으로 __init__을 nn.Module 것을 상속받게 한다. 또한 forward 함수를 통해 예측 y를 구하는 바를 해당 수식을 통해 가볍게 구현할 수 있다. 

저 포맷을 일반으로 다양한 모형 설계를 수행하기 때문에 저 틀을 알아놓고 가자.

 

그렇다면 이를 통해 선형 회귀를 구현해보자

 

이런 식으로 할 수 있다. 다중 선형 회귀도 마찬가지다.