인프런 커뮤니티 질문&답변

문환룡님의 프로필 이미지
문환룡

작성한 질문수

처음하는 딥러닝과 파이토치(Pytorch) 부트캠프 (쉽게! 기본부터 챗GPT 핵심 트랜스포머까지) [데이터분석/과학 Part4]

Binary Classfication 딥러닝 적용해보기 (URL Phishing Kaggle 문제)

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

작성

·

272

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() 안해도 됩니다.

감사합니다.

문환룡님의 프로필 이미지
문환룡

작성한 질문수

질문하기