ComeçarComece de graça

Usando TensorDataset

Estruturar seus dados em um conjunto de dados é uma das primeiras etapas do treinamento de uma rede neural do PyTorch. O site TensorDataset simplifica isso convertendo matrizes NumPy em um formato que o PyTorch pode usar.

Neste exercício, você criará um TensorDataset usando o conjunto de dados animals pré-carregado e inspecionará sua estrutura.

Este exercício faz parte do curso

Introdução à aprendizagem profunda com PyTorch

Ver curso

Instruções do exercício

  • Converta X e y em tensores e crie um TensorDataset.
  • Acesse e imprima a primeira amostra.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

import torch
from torch.utils.data import TensorDataset

X = animals.iloc[:, 1:-1].to_numpy()  
y = animals.iloc[:, -1].to_numpy()

# Create a dataset
dataset = ____(____, ____)

# Print the first sample
input_sample, label_sample = ____
print('Input sample:', input_sample)
print('Label sample:', label_sample)
Editar e executar o código