ComenzarEmpieza gratis

Escribir un bucle de entrenamiento

En scikit-learn el bucle de entrenamiento está envuelto en el método .fit() mientras que en PyTorch se configura manualmente. Aunque esto añade flexibilidad, requiere una implementación personalizada.

En este ejercicio, crearás un bucle para entrenar un modelo de predicción salarial.

La función show_results() te ayuda a visualizar algunas predicciones de ejemplo.

Las importaciones de paquetes proporcionadas son: pandas como pd, torch, torch.nn como nn, torch.optim como optim y DataLoader y TensorDataset desde torch.utils.data.

Se han creado las siguientes variables: num_epochs, que contiene el número de épocas (ajustado en 5); dataloader, que contiene el cargador de datos; model, que contiene la red neuronal; criterion, que contiene la función de pérdida, nn.MSELoss(); optimizer, que contiene el optimizador SGD.

Este ejercicio forma parte del curso

Introducción al aprendizaje profundo con PyTorch

Ver curso

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

# Loop over the number of epochs and then the dataloader
for i in ____:
  for data in ____:
    # Set the gradients to zero
    ____
Editar y ejecutar código