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
Instrucciones del ejercicio
- Crea una instancia de
WaterDataset
a partir dewater_train.csv
y asígnala adataset_train
. - Crea
dataloader_train
basado endataset_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)