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
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
____