다중 분류를 위한 Logistic Regression Model 만들기

앞서 포스팅 된 Logistic Regression 모델은 두 가지 Class에 대한 이진 분류를 수행하는것이다.

하지만 3개 이상의 Class 즉, 다중 분류를 위한 Logistic Regression은 어떻게 만들 수 있을까?

선형 결합 계층의 출력을 Class의 숫자만큼 설정하고,

손실함수로 Crossentropy에 Softmax가 추된 형태인 CrossEntropyLoss()를 이용한다.

아래의 코드와 같이 사용하면 손쉽게 다중분류문제도 해결해 볼 수 있다.