1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Nhập môn Deep Learning với PyTorch
Connected

Bài tập

Sử dụng MSELoss

Với bài toán hồi quy, bạn thường dùng Mean Squared Error (MSE) làm hàm loss thay vì cross-entropy. MSE tính bình phương chênh lệch giữa giá trị dự đoán (y_pred) và giá trị thực (y). Bây giờ, bạn sẽ tính MSE loss bằng cả NumPy và PyTorch.

Các gói torch, numpy (viết tắt np), và torch.nn (viết tắt nn) đã được import sẵn.

Hướng dẫn

100 XP
  • Tính MSE loss bằng NumPy.
  • Tạo hàm loss MSE bằng PyTorch.
  • Chuyển y_pred và y thành tensor, rồi tính MSE loss đặt tên là mse_pytorch.