ComeçarComece gratuitamente

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

Ver Curso

Instruções de exercício

  • Crie uma instância de WaterDataset a partir de water_train.csv, atribuindo-a a dataset_train.
  • Crie dataloader_train com base em dataset_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)
Editar e executar código