Como usar o TensorDataset
Estruturar seus dados em um conjunto de dados é uma das primeiras etapas do treinamento de uma rede neural do PyTorch. O TensorDataset simplifica isso convertendo matrizes NumPy em um formato que o PyTorch pode usar.
Neste exercício, você deve criar um TensorDataset usando o conjunto de dados animals pré-carregado e inspecionar sua estrutura.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Instruções do exercício
- Converta
Xeyem 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)