1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 텍스트 딥러닝
Connected

연습 문제

감성 분석 CNN 모델 테스트하기

모델을 학습했으니, PyBooks에서 새로운 도서 리뷰로 성능을 확인하려고 합니다.

각 리뷰의 감성이 긍정인지 부정인지 판별해야 합니다.

다음 패키지가 이미 임포트되어 있습니다: torch, torch.nn은 nn으로, torch.nn.functional은 F로, torch.optim은 optim으로.

또한 vocab_size와 embed_dim을 인수로 하는 TextClassificationCNN() 인스턴스가 로드되어 model로 저장되어 있습니다.

지침

100 XP
  • book_reviews 리스트를 순회하며, 각 리뷰의 단어들을 텐서로 변환하세요.
  • 각 input_tensor에 대해 모델의 출력을 구하세요.
  • outputs.data에서 가장 가능성이 높은 감성 범주의 인덱스를 찾으세요.
  • predicted_label 항목을 추출해 변환하고, 1이면 "Positive" 레이블의 감성 문자열로 바꾸세요.