ComenzarEmpieza gratis

DataLoader PyTorch

¡Buen trabajo! Has definido la clase de conjunto de datos. El WaterDataset que acabas de crear ya está disponible para que lo utilices.

El siguiente paso en la preparación de los datos de entrenamiento es configurar un DataLoader. Se puede crear un PyTorch DataLoader a partir de un Dataset para cargar datos, dividirlos en lotes y realizar transformaciones en los datos si se desea. A continuación, obtiene una muestra de datos lista para el entrenamiento.

En este ejercicio, debes construir un DataLoader basado en el WaterDataset. La clase DataLoader ya se ha importado para ti desde torch.utils.data. ¡Manos a la obra!

Este ejercicio forma parte del curso

Aprendizaje profundo intermedio con PyTorch

Ver curso

Instrucciones del ejercicio

  • Crea una instancia de WaterDataset a partir de water_train.csv y asígnala a dataset_train.
  • Crea dataloader_train basado en dataset_train, utiliza un tamaño de lote de dos y baraja las muestras.
  • Obtén un lote de características y etiquetas del DataLoader e imprímelas.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

# Create an instance of the WaterDataset
dataset_train = ____(____)

# Create a DataLoader based on dataset_train
dataloader_train = ____(
    ____,
    batch_size=____,
    shuffle=____,
)

# Get a batch of features and labels
features, labels = ____
print(features, labels)
Editar y ejecutar código