ComenzarEmpieza gratis

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

Ver curso

Instrucciones del ejercicio

  • Convierte X e y en tensores y crea un TensorDataset.
  • 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)
Editar y ejecutar código