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 do exercício
- Crie uma instância de
WaterDataseta partir dewater_train.csv, atribuindo-a adataset_train. - Crie
dataloader_traincom 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 completando 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)