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
Instruções do exercício
- Converta
X
ey
em tensores e crie umTensorDataset
. - 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)