• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    미해결

Binary Classfication 딥러닝 적용해보기 질문입니다

23.02.27 22:27 작성 조회수 220

0

Binary Classfication 딥러닝 적용해보기 강의 19:13초에서,

y_pred_list.extend(y_test_pred.squeeze().detach().tolist())

를 실행한뒤, y_pred_list = torch.tensor(y_pred_list).unsqueeze(1)을 실행하는데,

y_pred_list의 size() 를 y_test_tensor의 size()와 맞추는거면

predlist.extend(y_test_pred.detach().tolist())

를 실행한뒤, y_pred_list = torch.tensor(y_pred_list)) 을 실행해도 같은 건가요?

답변 1

답변을 작성해보세요.

0

안녕하세요. 답변도우미입니다.

아핫 좀더 공식적으로 shape 을 만드는 부분을 보여드리려 하고, 관련 함수도 좀더 익숙해질겸 위 코드를 작성하였고요. 말씀하신대로 squeeze(), unsqeeze() 안해도 됩니다.

감사합니다.