1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch로 배우는 딥러닝 입문
Connected

演習

MSELoss 사용하기

회귀 문제에서는 교차 엔트로피 대신 Mean Squared Error(MSE)를 손실 함수로 자주 사용해요. MSE는 예측값(y_pred)과 실제값(y)의 제곱 차이를 계산합니다. 이제 NumPy와 PyTorch를 사용해 MSE 손실을 각각 계산해 볼게요.

torch, numpy(별칭 np), torch.nn(별칭 nn) 패키지는 이미 임포트되어 있어요.

指示

100 XP
  • NumPy로 MSE 손실을 계산하세요.
  • PyTorch로 MSE 손실 함수를 생성하세요.
  • y_pred와 y를 텐서로 변환한 뒤, MSE 손실을 mse_pytorch로 계산하세요.