Aan de slagGa gratis aan de slag

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

Cursus bekijken

Oefeninstructies

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

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in 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