딥러닝 파이토치 교과서 2.4장 오류있지 않나요?

자유게시판
1

car_evaluation.csv (50.7 KB)

ipynb가 업로드 안 되는 점 양해 구합니다.

제가 이 커뮤니티에서 입문 저서로 서지영님의 <딥러닝 파이토치 교과서>를 알게 되고 공부 중인데요.

코드 2-11에서

outputs = pd.get_dummies(dataset.output)
outputs = outputs.values
outputs = torch.tensor(outputs).flatten()

이렇게 되어 있는데 여기서 dummies를 쓰면 안 된다고 생각합니다.
이러면 outputs에 0과 1만 들어가죠.

그런데 코드 2-16에서 print(y_val[:5])하면 교재에 나와있듯이

tensor([[2.7215, 1.6601, -2.2784, -2.1693],
        [2.8467, 1.7512, -2.3375, -2.2104],
         ....                             ])

이렇게 나오고 y_val = np.argmax(y_val, axis = 1) 하면 0~3까지 나올 수 있죠.

입문 저서의 입문 파트에서 이래 버리니 정말 환장하겠네요.

혹시 이 책 있으신 분 답변해주시면 정말 감사하겠습니다.

2개의 좋아요
2

현재 올려주신 코드만으로는 정확한 원인을 판단하기 어렵네요. 특히 outputs는 정답 레이블이고 y_val은 모델의 예측값일 수 있으므로, 두 값의 형태가 다르다는 이유만으로 get_dummies()가 잘못되었다고 판단할 수는 없습니다.