-
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
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)) 을 실행해도 같은 건가요?
처음하는 딥러닝과 파이토치(Pytorch) 부트캠프 [데이터과학 Part3]
Binary Classfication 딥러닝 적용해보기 (URL Phishing Kaggle 문제)
강의실 바로가기
답변을 작성해보세요.
0
잔재미코딩 DaveLee
지식공유자2023.02.28
안녕하세요. 답변도우미입니다.
아핫 좀더 공식적으로 shape 을 만드는 부분을 보여드리려 하고, 관련 함수도 좀더 익숙해질겸 위 코드를 작성하였고요. 말씀하신대로 squeeze(), unsqeeze() 안해도 됩니다.
감사합니다.
답변 1