1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Introduction to Deep Learning with PyTorch
Connected

cvičení

Použití MSELoss

Pro regresní úlohy se jako ztrátová funkce často používá střední kvadratická chyba (MSE) místo křížové entropie. MSE počítá kvadratický rozdíl mezi předpovězenými hodnotami (y_pred) a skutečnými hodnotami (y). Teď vypočítáš MSE loss pomocí NumPy i PyTorche.

Balíčky torch, numpy (jako np) a torch.nn (jako nn) jsou už naimportované.

Pokyny

100 XP
  • Vypočítej MSE loss pomocí NumPy.
  • Vytvoř ztrátovou funkci MSE pomocí PyTorche.
  • Převeď y_pred a y na tensory a pak vypočítej MSE loss jako mse_pytorch.