CommencezCommencez gratuitement

Utiliser TensorDataset

Structurer vos données dans un ensemble de données est l'une des premières étapes pour entraîner un réseau de neurones PyTorch. TensorDataset simplifie cela en convertissant des tableaux NumPy en un format utilisable par PyTorch.

Dans cet exercice, vous allez créer un TensorDataset à partir de l'ensemble de données animals déjà chargé et examiner sa structure.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Instructions de l’exercice

  • Convertissez X et y en tenseurs et créez un TensorDataset.
  • Accédez à l'échantillon initial et affichez-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