CommencerCommencez gratuitement

Utilisation de TensorDataset

La structuration de vos données en un ensemble de données est l’une des premières étapes de l’apprentissage d’un réseau neuronal PyTorch. TensorDataset simplifie cette opération en convertissant des tableaux NumPy dans un format utilisable par PyTorch.

Dans cet exercice, vous allez créer un TensorDataset en utilisant le jeu de données préchargé animals et inspecter sa structure.

Cet exercice fait partie du cours

<cours>Introduction au deep learning avec PyTorch</cours>
Voir le cours

Instructions de l’exercice

  • Convertissez X et y en tenseurs et créez un TensorDataset.
  • Accédez au premier échantillon et imprimez-le.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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)
Modifier et exécuter le code