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

연습 문제

텍스트용 CNN 모델 학습

TextClassificationCNN 클래스를 잘 정의하셨어요. 이제 PyBooks에서 모델을 학습해 도서 리뷰의 감성 분석 정확도를 높이려고 해요.

다음 패키지는 미리 임포트되어 있어요: torch, torch.nn은 nn으로, torch.nn.functional은 F로, torch.optim은 optim으로.

또한 vocab_size와 embed_dim을 인수로 하는 TextClassificationCNN() 인스턴스를 로드해 model로 저장해 두었어요.

지침

100 XP
  • 이진 분류에 사용하는 손실 함수를 정의하고 criterion으로 저장하세요.
  • 학습 루프 시작 시 그래디언트를 0으로 초기화하세요.
  • 루프 끝에서 파라미터를 업데이트하세요.