1. Обучение
    2. /
  2. Курса
    3. /
  3. Введение в глубокое обучение с PyTorch
Connected

упражнение

Использование MSELoss

В задачах регрессии вместо кросс-энтропии часто используют среднеквадратичную ошибку (MSE) в качестве функции потерь. MSE вычисляет квадрат разности между предсказанными значениями (y_pred) и фактическими (y). В этом упражнении вы вычислите потери MSE двумя способами: с помощью NumPy и PyTorch.

Пакеты torch, numpy (как np) и torch.nn (как nn) уже импортированы.

Инструкции

100 XP
  • Вычислите потери MSE с помощью NumPy.
  • Создайте функцию потерь MSE с помощью PyTorch.
  • Преобразуйте y_pred и y в тензоры, затем вычислите потери MSE и сохраните результат в переменную mse_pytorch.