1. Nauka
    2. /
  2. Kursy
    3. /
  3. Wprowadzenie do uczenia głębokiego z PyTorch
Connected

ćwiczenie

Użycie funkcji MSELoss

W problemach regresji często używa się błędu średniokwadratowego (MSE) jako funkcji straty zamiast entropii krzyżowej. MSE oblicza kwadrat różnicy między wartościami przewidywanymi (y_pred) a rzeczywistymi (y). Teraz obliczysz stratę MSE, korzystając zarówno z NumPy, jak i PyTorch.

Pakiety torch, numpy (jako np) oraz torch.nn (jako nn) są już zaimportowane.

Instrukcje

100 XP
  • Oblicz stratę MSE za pomocą NumPy.
  • Utwórz funkcję straty MSE za pomocą PyTorch.
  • Przekonwertuj y_pred i y na tensory, a następnie oblicz stratę MSE jako mse_pytorch.