Aan de slagBegin gratis

TensorDataset gebruiken

Je data structureren in een dataset is een van de eerste stappen bij het trainen van een PyTorch-neuraal netwerk. TensorDataset maakt dit eenvoudiger door NumPy-arrays om te zetten naar een formaat dat PyTorch kan gebruiken.

In deze oefening maak je een TensorDataset met de vooraf geladen animals-gegevensset en bekijk je de structuur ervan.

Deze oefening maakt deel uit van de cursus

Introductie tot Deep Learning met PyTorch

Bekijk cursus

Oefeninstructies

  • Zet X en y om naar tensors en maak een TensorDataset.
  • Benader en print de eerste sample.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

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)
Code bewerken en uitvoeren