더북(TheBook)

① 클래스(class) 형태로 구현되는 모델은 nn.Module을 상속받습니다.

__init__()은 모델에서 사용될 파라미터와 신경망을 초기화하기 위한 용도로 사용하며, 객체가 생성될 때 자동으로 호출됩니다. __init__()에서 전달되는 매개변수는 다음과 같습니다.

self: 첫 번째 파라미터는 self를 지정해야 하며 자기 자신을 의미합니다. 예를 들어 ex라는 함수가 있을 때 self 의미는 다음 그림과 같습니다.

▲ 그림 2-31 self 의미

embedding_size: 범주형 칼럼의 임베딩 크기

output_size: 출력층의 크기

layers: 모든 계층에 대한 목록

p: 드롭아웃(기본값은 0.5)

super().__init__()은 부모 클래스(Model 클래스)에 접근할 때 사용하며 superself를 사용하지 않는 것에 주의해야 합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.