Utilizar TensorDataset
Estructurar tus datos en un conjunto de datos es uno de los primeros pasos para entrenar una red neuronal PyTorch. TensorDataset
lo simplifica convirtiendo las matrices NumPy en un formato que PyTorch pueda utilizar.
En este ejercicio, crearás un TensorDataset
utilizando el conjunto de datos precargado animals
e inspeccionarás su estructura.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Instrucciones del ejercicio
- Convierte
X
ey
en tensores y crea unTensorDataset
. - Accede a la primera muestra e imprímela.
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
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)