1. 학습
    2. /
  2. 강의
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

연습 문제

예측 모델 평가하기

이제 평가해 볼 시간이에요! 이전 연습 문제에서 학습한 것과 같은 LSTM 네트워크가 몇 에포크 더 학습되어 net으로 제공돼요.

여러분의 과제는 테스트 데이터셋에서 평균제곱오차(Mean Squared Error) 지표를 사용해 모델을 평가하는 거예요 (torchmetrics는 이미 임포트되어 있어요). 모델이 얼마나 잘 작동하는지 확인해 볼까요?

지침

100 XP
  • 평균제곱오차(Mean Squared Error) 지표를 정의하고 mse에 할당하세요.
  • 입력 시퀀스를 net에 전달하고, 결과를 outputs에 할당하기 전에 squeeze로 차원을 줄이세요.
  • 테스트 지표의 최종 값을 계산해 test_mse에 할당하세요.