DataLoader do PyTorch
Bom trabalho ao definir a classe Dataset! O WaterDataset
que você acabou de criar agora está disponível para uso.
A próxima etapa na preparação dos dados de treinamento é configurar um DataLoader
. Um PyTorch DataLoader
pode ser criado a partir de um Dataset
para carregar dados, dividi-los em lotes e realizar transformações nos dados, se desejado. Em seguida, ele produz uma amostra de dados pronta para treinamento.
Neste exercício, você criará um DataLoader
com base no WaterDataset
. A classe DataLoader
de que você precisará já foi importada para você de torch.utils.data
. Vamos lá!
Este exercício faz parte do curso
Aprendizagem profunda intermediária com PyTorch
Instruções de exercício
- Crie uma instância de
WaterDataset
a partir dewater_train.csv
, atribuindo-a adataset_train
. - Crie
dataloader_train
com base emdataset_train
, usando um tamanho de lote de dois e embaralhando as amostras. - Obtenha um lote de recursos e etiquetas do DataLoader e imprima-os.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# 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)